[프로그래머스] 카펫

문제링크

다시 풀어보기…

파이썬

def solution(brown, yellow):
    answer = []
    t = brown + yellow
    for i in range(1,t+1):
        if (t/i) %1 == 0:
            a = t/i
            if a>=i:
                if 2*a + 2*i == brown+4:
                    return [a,i]
    return answer