[Python] 프로그래머스 42842. 카펫 (Lv.2)

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

프로그래머스

 

규칙을 찾으면 풀리는 간단한 완전탐색 문제입니다. 옛날 문제라 레벨이 높게 책정된것 같네요.

 

카펫의 가로길이가 세로길이 이상이라는 조건이 주어져있기 때문에, 가능한 경우에 대해 테스트해보면 됩니다.

가로길이가 w 세로길이가 h라고 하면 갈색 카펫의 수는 2 * (w + h) - 4이고 노란 카펫의 수는 (w - 2) * (h - 2)입니다.

w >= h인 가능한 모든 w와 h에 대해 탐색해도 내가 시도해봐야할 w와 h 조합이 대략 w^2개이고 w는 2500 이하이기때문에 그렇게 긴 시간이 걸리지 않습니다.

 

def solution(brown, yellow):
    for i in range(2, brown):
        for j in range(2, brown):
            if j > i:
                break
            if brown == 2 * i + 2 * j - 4 and yellow == (i - 2) * (j - 2):
                return [i, j]