코딩테스트

프로그래머스 카펫

박준희 2021. 11. 18. 13:52

https://programmers.co.kr/learn/courses/30/lessons/42842

 

코딩테스트 연습 - 카펫

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

programmers.co.kr

 

def solution(brown, yellow):
    answer = []
    for w in range(1, yellow + 1):
        print(w)
        if yellow % w != 0:
            continue
        h = yellow / w
        
        if brown == (w+2) * (h+2) - yellow:
            if (w+2) >= (h+2):
                answer.append(w+2)
                answer.append(h+2)
            else:
                answer.append(h+2)
                answer.append(w+2)
            break
        
    return answer