[KOTIN] 카펫

- https://programmers.co.kr/learn/courses/30/lessons/42842

코딩테스트 연습 - 카펫

문제 설명

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

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

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

제한 사항

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

입출력 예

brown	yellow	return
10	2	[4, 3]
8	1	[3, 3]
24	24	[8, 6]

풀이

1. 노란 격자의 가로/세로 값을 구하는 탐색을 한다.

2. 1번에서 나온 노란격자의 크기를 계산해서 갈색 격자와 같은 값을 결과로 구한다.

* 전체 영역을 먼저 구한 뒤 가로 에서 2를 빼고 세로에서 2를 뺏을 때 노란 격자의 개수와 같은 지 확인하면 알고리즘이 더 간편해 진다.

전체 코드

class Solution {
    fun solution(brown: Int, yellow: Int): IntArray {
        var answer = IntArray(2)
        var yellowWidth = 0
        var yellowHeight = 0
        var index = yellow
        
        // yellow 개수를 절반으로 계속 나눠서 brown 과 같을 개수인 것을 찾는다.
        while (true) {
            // 나누어 떨어지는 값 찾기
            while (yellow % index != 0) {
                index--
            }

            //println("test : $index, $yellow")
            // 노란 격자의 가로/세로 구하기
            yellowHeight = yellow / index
            yellowWidth = yellow / yellowHeight
            //println("test2 : $yellowWidth, $yellowHeight")
                        
            // 갈색 격자와 같을 경우 결과 출력
            if ((yellowWidth + 2) * 2 + yellowHeight * 2 == brown) break
            else index--
            
        }
        
        answer[0] = yellowWidth + 2
        answer[1] = yellowHeight + 2
        
        return answer
    }
}