[프로그래머스 Lv.2] 카펫 (kotlin ver.)

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

프로그래머스

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

        var list = mutableListOf<Int>()
        val sum = brown + yellow

        for (i in 3..sum) {
            if (sum % i == 0) list.add(i)
        }


        list.forEach {
            val row = it
            var col = yellow / it

            if ((row + 2) * (col + 2) == sum) {
                answer = intArrayOf(row, col)
            }
        }

        return answer
    }
}

그림을 자세히 보면 갈색블럭은 노란색 블럭의 테두리를 무조건 감싸고 있다는 점과

리턴 값은 항상 총 블럭 개수인 갈색 블럭과 노란색 블럭의 합의 약수이다는 점에 주목하면 된다.

 

따라서 위 코드는 약수들을 구한 다음 그 약수들 중에 갈색 블럭이 노란색 블럭보다 한변에 2개씩 많다는 점을 이용한 코드다.