프로그래머스 카펫

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

코딩테스트 연습 - 카펫

 

def solution(brown, yellow):
    answer = []
    for w in range(1, yellow + 1):
        print(w)
        if yellow % w != 0:
            continue
        h = yellow / w
        
        if brown == (w+2) * (h+2) - yellow:
            if (w+2) >= (h+2):
                answer.append(w+2)
                answer.append(h+2)
            else:
                answer.append(h+2)
                answer.append(w+2)
            break
        
    return answer