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

문제 링크: https://school.programmers.co.kr/learn/courses/30/lessons/42842

프로그래머스

풀이

문제 해결을 위해 성립해야 할 조건은 아래와 같다.

1. 카펫의 가로, 세로의 곱은 카펫의 총 넓이이기에 갈색과 노란색 부분을 합친 넓이이다.
2. 가장자리를 제외한 카펫의 넓이, 즉 가로와 세로에서 2를 뺀 값들의 곱은 카펫의 노란색 부분과 같다.

완전 탐색으로 위의 조건들을 체크하면서 모든 조건을 만족하는 가로와 세로의 길이를 찾아가면 된다.

def solution(brown, yellow):
    total = brown + yellow # 총 카펫의 크기
    for c in range(brown, 0, -1):
        if total % c != 0: continue # 나누어 떨어지지 않아 곱으로 나타낼 수 없는 경우는 스킵
        r = total // c # 가로와 세로의 곱은 전체 카펫의 넓이
        
        if c > 2 and r > 2 and (c-2)*(r-2) == yellow: # 가장 자리를 제외한 부분이 노란색 부분과 동일할 경우
            return [c, r]