def solution(brown, yellow):
width = yellow
height = 1
while height <= width:
# print(width, height)
if ((width+2)*2 + 2*height)==brown:
return [width+2, height+2]
height +=1
if yellow % height == 0:
width = yellow/height
어렵다고 생각했는데 생각보다 간단했다.
노란색 부분의 height를 1부터 늘려가면서 최적을 구하면 되었다.