[코딩테스트 입문] 2차원으로 만들기

[문제 설명]

정수 배열 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	result
[1, 2, 3, 4, 5, 6, 7, 8]	2	[[1, 2], [3, 4], [5, 6], [7, 8]]

 

[제한사항]

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

 

[입출력 예]

num_list	n	result
[1, 2, 3, 4, 5, 6, 7, 8]	2	[[1, 2], [3, 4], [5, 6], [7, 8]]
[100, 95, 2, 4, 5, 6, 18, 33, 948]	3	[[100, 95, 2], [4, 5, 6], [18, 33, 948]]

 

[입출력 예 설명]

입출력 예 #1

• num_list가 [1, 2, 3, 4, 5, 6, 7, 8] 로 길이가 8이고 n이 2이므로 2 * 4 배열로 변경한 [[1, 2], [3, 4], [5, 6], [7, 8]] 을 return합니다.

입출력 예 #2

• num_list가 [100, 95, 2, 4, 5, 6, 18, 33, 948] 로 길이가 9이고 n이 3이므로 3 * 3 배열로 변경한 [[100, 95, 2], [4, 5, 6], [18, 33, 948]] 을 return합니다.

 

 

 

[문제 풀이]

class Solution {
    public int[][] solution(int[] num_list, int n) {
        int x = num_list.length/n;
        int cnt = 0;
        int[][] answer = new int[x][n];
        
        
        // 배열 값을 2차원 배열로 만들기
        for(int j=0; j<x; j++) {
            for(int k=0; k<n; k++) {
                answer[j][k] = num_list[cnt];
                cnt++;
            }
        }
        
        return answer;
    }
}

 

이중 for문을 통해 answer의 2차원 배열에 값을 넣을 수 있다.

배열에 값을 넣을 때에는 for문을 이용하는 것 보다는 cnt++을 통해 하나씩 값을 넣어줄 수 있다.