프로그래머스 카펫(python)

def solution(brown, yellow):
    for i in range(1, int(yellow ** (1/2))+1):
        if yellow % i == 0:
            a = i #세로
            b = yellow // i #가로
            if brown == (a * 2 + (b + 2) * 2):               
                return [b+2,a+2]