[완전탐색] 카펫

코딩테스트 연습 - 카펫

import math
 
def solution(brown, yellow):
 
    def cal(num):
        temp = []
        for i in range(1, int(math.sqrt(num)) + 1):
            if num % i == 0:
                y = num // i
                temp.append([y, i])
        return temp
 
    answer = []
    result = cal(yellow)
    for x, y in result:
        if brown == (x + 2 + y) * 2:
            return [x + 2, y + 2]
    return answer
 
 
print(solution(24, 24))

---

solution(24, 24) 기준 temp = [[24, 1], [12, 2], [8, 3], [6, 4]]

넓이가 yellow가 될 수 있는 모든 경우의 수를 담은 temp를 반환하는 cal함수를 만들었다.
int(math.sqrt(num)) + 1는 자주 사용하는 방식이니 꼭 숙지하자.

내부 사각형 yellow의 가로 길이는 x, 세로 길이는 y이다.
이 (x, y) 조합을 하나씩 확인하여 brown이 될 수 있는 조건을 충족시키는 (x, y)를 찾고, 찾으면 +2 씩 한 값을 반환해준다.