[프로그래머스] lv.2 카펫 풀이

문제

https://school.programmers.co.kr/learn/courses/30/lessons/42842

프로그래머스

 

문제 풀이

가능한 모든 경우의 노란 직사각형 모양을 검사함.

yw, yh를 구해서 그 넓이를 둘러싼 길이가 brown과 같으면 True 리턴.

 

코드

def solution(brown, yellow):
    res = []
    for yw in range(1, yellow+1): 
        yh = yellow/yw
        if yh%1!=0 and yw*yh!=yellow:
            continue
        
        if (yw+yh+4)*2-4 == brown:
            res = [yw+2, yh+2]
            return sorted(res, reverse=True)