프로그래머스 완전탐색 '카펫' 파이썬 풀이

프로그래머스

"""
Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 노란색으로 칠해져 있고 
테두리 1줄은 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.

Leo는 집으로 돌아와서 아까 본 카펫의 노란색과 갈색으로 
색칠된 격자의 개수는 기억했지만, 전체 카펫의 크기는 기억하지 못했습니다.

Leo가 본 카펫에서 갈색 격자의 수 brown, 노란색 격자의 수 yellow가 
매개변수로 주어질 때 카펫의 가로, 세로 크기를 순서대로 
배열에 담아 return 하도록 solution 함수를 작성해주세요.
"""

def solution(b, y):
    # b+y = w*h
    # (w-2)*(h-2) = y
    
    for i in range(3, b+y):
        if ((i-2)*((b+y)/i -2)) == y:
            return sorted([i, (b+y)/i], reverse=True)

"""
(result[0] = w, result[1] = h)
표를 보니까 brown과 yellow를 더한값이 w*h였고,
(w-2)*(h-2) = yellow였다.

연립방정식을 풀어서 맞다면 더 큰값이 w가 되게 소트해서 리턴해주었다.
"""