카펫 - 프로그래머스

카펫 - 프로그래머스

def solution(brown, yellow):
    for i in range(1, int(yellow ** 0.5) + 1):
        if yellow % i == 0:
            j = yellow // i
            if 2*(i + j) + 4 == brown:
                return [j + 2, i + 2]