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의 배 수개입니다.
• 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]]

풀이

• 2차원 배열을 반환하는 문제
• 행의 길이는 주어진 배열의 총 길이를 n으로 나눈 값이고, 열의 길이는 n
• 행을 바꿔가면서 열을 하나씩 채워야 하기 때문에 이중반복문 사용
• 주어진 배열에 차례대로 접근하여 대입해야 하기 때문에 count 변수 선언
• 다만, 조금 더 생각을 해보면 몫과 나머지를 이용하여 단일 반복문을 사용할 수 있다

코드

class Solution {
    public int[][] solution1(int[] num_list, int n) {
        int[][] answer = new int[num_list.length/n][n];
        int count = 0;
        for(int i = 0; i < num_list.length/n; i++){
            for(int j = 0; j < n; j++){
                answer[i][j] = num_list[count];
                count++;
            }
        }
        return answer;
    }
    
    public int[][] solution2(int[] num_list, int n){
    	int length = num_list.length;
        int[][] answer = new int[length/n][n];

        for (int i = 0; i < length; i++) {
            answer[i/n][i%n] = num_list[i];
        }
        return answer;
    }
}