[programmers] 카펫

---

로직

brown + yellow로 가능한 곱 조합을 나타낸다.

이후 for문을 돌면서 양옆, 위아래 1개씩 (총 2개씩) 빼서 곱한 값이 yellow면 답이다

 

코드

def solution(brown, yellow):
    total = brown + yellow
    possible = []
    for i in range(total, 2, -1):
        if total % i == 0 and i >= total // i:
            possible.append([i, total // i])

    for i in possible:
        if (i[0] - 2) * (i[1] - 2) == yellow:
            return i