[프로그래머스/python] 코딩테스트 연습 Lv.2 카펫

문제

입출력 예

 

Solution

def solution(brown, yellow):

    y = int(brown/2) - 2
    for i in range(1, y):
        if yellow == (y-i)*i:
            return [y-i+2, i+2]