[Programmers] level2 - 카펫 (Python) : 완전 탐색

▶카펫 - 완전 탐색 (level 2)

▶문제

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

Leo는 집으로 돌아와서 아까 본 카펫의 노란색과 갈색으로 색칠된 격자의 개수는 기억했지만, 전체 카펫의 크기는 기억하지 못했습니다.

Leo가 본 카펫에서 갈색 격자의 수 brown, 노란색 격자의 수 yellow가 매개변수로 주어질 때 카펫의 가로, 세로 크기를 순서대로 배열에 담아 return 하도록 solution 함수를 작성해주세요.

 

▶제한사항

• 갈색 격자의 수 brown은 8 이상 5,000 이하인 자연수입니다.
• 노란색 격자의 수 yellow는 1 이상 2,000,000 이하인 자연수입니다.
• 카펫의 가로길이는 세로 길이와 같거나, 세로 길이보다 깁니다.

 

▶입출력 예

 

▶풀이

brown의 가로, 세로의 길이는 yellow의 가로, 세로보다 2 크다.

넓이를 보면 (yellow_x + 2) * (yellow_y + 2) = brown + yellow가 된다.

 

그리고 1 ~ yellow까지 반복문을 돌리면서 yellow로 i가 나눠지는지 확인 후에

y_x, y_y에 yellw // i와 i를 대입해준다.

그 값을 가지고 위 넓이를 구한 식에 대입하면 정답이 나온다.

def solution(brown, yellow):
    answer = []
    y_x, y_y = 0, 0
    
    for i in range(1, yellow + 1):
        if yellow % i != 0:
            continue
        y_x = yellow // i
        y_y = i
        
        if (y_x + 2) * (y_y + 2) == brown + yellow:
            b_x = y_x + 2
            b_y = y_y + 2
            answer = [b_x, b_y]
            return answer
            
    return answer