세용용용용 2023. 5. 23. 08:32

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)