[프로그래머스 코딩테스트 고득점 kit] 카펫 python

def solution(brown, yellow):
    answer = []
    sum = yellow + brown
    for i in range(3, (sum//2 + 1)):
        if sum % i == 0:
            if 2*(sum//i + i) - 4 == brown:
                answer.append(sum//i)
                answer.append(i)
                break

    return answer