[Algorithm] 프로그래머스 - 정수를 나선형으로 배치하기 (Kotlin)

양의 정수 n이 매개변수로 주어진다. n x n 배열에 1부터 n^2까지 정수를 인덱스 [0][0]부터 시계방향 나선형으로 배치한 2차원 배열을 리턴하는 함수를 작성하라

 

 

 

class Solution {
    fun solution(n: Int): Array<IntArray> {
        val result = Array(n) { IntArray(n) }
        var value = 1
        var x = 0
        var y = 0
        var direction = 0

        val dx = arrayOf(0, 1, 0, -1)
        val dy = arrayOf(1, 0, -1, 0)

        while (value <= n * n) {
            result[x][y] = value
            value++
            
            var nx = x + dx[direction]
            var ny = y + dy[direction]

            if (nx !in 0 until n || ny !in 0 until n || result[nx][ny] != 0) {
                direction = (direction + 1) % 4
                nx = x + dx[direction]
                ny = y + dy[direction]
            }

            x = nx
            y = ny
        }

        return result
    }
}