[Programmers] [완전탐색] [Python] Level2_카펫

문제

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

프로그래머스

풀이

# 프로그래머스 2단계 - 완전탐색 - 카펫
def solution(brown, yellow):
    answer = []
    total = brown + yellow
    for weight in range(total, 2, -1):
        if total % weight == 0:
            height = total // weight
            if yellow == (weight-2) * (height-2):
                return [weight, height]

 

일단 처음에 문제를 풀려고 할때 total 의 약수를 구해서

중간배열의 값을 return 하면 될줄 알았지만 코드를 구현해본 결과 그렇지 않았다.

조금 더 생각해보았는데 

카펫의 넓이 : weight * height

외부 사각형의 갯수(brown) = 2w + 2h - 4 

내부 사각형의 갯수(yellow) = (w-2) * (h-2)

여기서 수학적으로 접근하지 않아서 조금 생각을 많이 하였는데 

brown = 2w + 2h - 4 인 이유가 모서리를 제외하고 brown카펫이 yellow카펫보다 위 아래로 한칸씩 더 많이 떄문에 

위에 브라운 가로 카펫 w, 아래 브라운 가로 카펫 w, 위에 브라운 세로 카펫 h, 아래 브라운 가로 카펫 h

이렇게 되어 모서리는 겹치는 부분이기 때문에 4개를 제외하면 2w + 2h -4 라는 값이 나오게 된다.

또한 yellow 카펫은 brown 카펫보다 가로 세로가 2칸씩 적기 떄문에 -2를 해준다. 

 total(brown+yellow)이 weight로 나누어진다면 height가 몫이 될 것이다. 

따라서 그 숫자들을 비교해 가면서 (w-2) * (h-2) 가 yellow 값이 된다면 그 값을 반환해도록 해준다.