[프로그래머스/Kotlin] 2차원으로 만들기 (Lv. 0)

문제

정수 배열 num_list와 정수 n이 매개변수로 주어집니다. num_list를 다음 설명과 같이 2차원 배열로 바꿔 return하도록 solution 함수를 완성해주세요.

num_list가 [1, 2, 3, 4, 5, 6, 7, 8] 로 길이가 8이고 n이 2이므로 num_list를 2 * 4 배열로 다음과 같이 변경합니다. 2차원으로 바꿀 때에는 num_list의 원소들을 앞에서부터 n개씩 나눠 2차원 배열로 변경합니다.

 

제한사항

• num_list의 길이는 n의 배 수개입니다.
• 0 ≤ num_list의 길이 ≤ 150
• 2 ≤ n < num_list의 길이

 

입출력 예

 

코드

class Solution {
    fun solution(num_list: IntArray, n: Int): Array<IntArray> {
        var answer: Array<IntArray> = arrayOf<IntArray>()
        
        var rows = num_list.size / n
        var cols = n
        var array = Array(rows){ IntArray(cols) }
        var idx = 0
        for(r in 0 until rows){
            for(c in 0 until cols){
                array[r][c] = num_list[idx++]
            }
        }
        return array
    }
}

 

기억하고 싶은 부분

• 2차원 배열 만들기

var arry = Array( 행의 크기 ) { IntArray( 열의 크기 ) }

 

 

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

프로그래머스