[Programmers] LV2. 카펫

문제

---

Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 노란색으로 칠해져 있고 테두리 1줄은 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.

 

Leo는 집으로 돌아와서 아까 본 카펫의 노란색과 갈색으로 색칠된 격자의 개수는 기억했지만, 전체 카펫의 크기는 기억하지 못했습니다. Leo가 본 카펫에서 갈색 격자의 수 brown, 노란색 격자의 수 yellow가 매개변수로 주어질 때 카펫의 가로, 세로 크기를 순서대로 배열에 담아 return 하도록 solution 함수를 작성해 주세요.

 

제한사항 

- 갈색 격자의 수 brown은 8 이상 5,000 이하인 자연수입니다.

- 노란색 격자의 수 yellow는 1 이상 2,000,000 이하인 자연수입니다.

- 카펫의 가로길이는 세로 길이와 같거나, 세로 길이보다 깁니다.

 

풀이

---

우리에게 주어진 값은 Yellow, Brown의 수이다. 그러면 단순하게 카펫의 가로를 i, 세로를 j라고 했을 때 (i-2) * (j-2) = yellow가 된다. 왜냐하면 가장자리의 값들은 항상 양옆으로 2개씩 있으니깐! 이를 고려해서 i값을 키워가며 조건이 맞았을 때의 가로 i와 세로 j를 return 해주면 된다. 이때, 제한사항에 있는 i >= j도 고려해 줘야 한다.

 

def solution(brown, yellow):
    
    total = brown + yellow
    
    for i in range(1, 2000000):
        if total % i == 0: # 가로
            j = total / i # 세로
            if (i-2)*(j-2) == yellow and i >= j:
                return [i, j]