[Python] lv.2 카펫 | 완전탐색

코드:

def solution(brown, yellow):

    total = brown + yellow                  # a * b = total
    for b in range(1,total+1):
        if total % b == 0:            # total / b = a
            a = total / b
            if a >= b:                      # a >= b
                if 2*a + 2*b == brown + 4:  # 2*a + 2*b = brown + 4 
                    return [a,b]

 

풀이:

가로를 a, 세로를 b 라고 하고 수식 세우기