카펫

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

나의 코드

def solution(brown, yellow):

    answer=[]

    for i in range(1,int(yellow**(1/2))+1):

        if yellow%i == 0 and 2*(i+(yellow//i))+4==brown:

            answer = [(yellow//i)+2, i+2]

    print(answer)

    return answer

solution(24,24)