[프로그래머스_Kotlin] 카펫

프로그래머스

 

문제

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

 

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

 

 

제한사항

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

 

 

입출력 예

 

 

 

 

 

내 풀이

일단 문제를 이해하는데 오래걸렸다.

이해라기보다는 규칙을 찾는데 오래걸렸다.

 

먼저 생각한 것은

yellow 1 -> 3X3 (brown 8)
가로 >= 세로 (yellow가 가로로 나열 / 정사각형)

 

그럼 yellow가 4개일 경우 정사각형과 가로로 나열되는 것 두가지가 존재하지 않나?

이렇게 되면 규칙을 정사각형이 되는 경우와 가로로 나열된 부분을 따로 정해야하나?

라고 생각했다.

 

위에 그림을 설명하자면

먼저 yellow가 1개면 3X3이 된다.

2개면 4X3, 3개면 5X3

내가 생각한 4개일때 4X4 6X3 두개의 경우가 있다.

 

하지만 너무 깊게 생각했다.

총 갯수의 (가로-2) * (세로-2)를 하면 yellow의 크기가 나타난다!...

3 -2 X 3 -2= (1.1)

4 X 3 = (2.1)

...

class Solution {
    fun solution(brown: Int, yellow: Int): IntArray {
        var answer = intArrayOf()
        val all = brown + yellow

        for (y in 3..all / 3) {
            val x = all / y
            if ((x - 2) * (y - 2) == yellow) {
                answer += x
                answer += y
                break
            }
        }
        return answer
    }
}