programmers

[JAVA] 2차원으로 만들기 다차원배열

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

* 입출력 예 #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[갯수][길이]
        int[][] answer = new int[num_list.length/n][n];
        int a = 0;
        for(int i=0; i<num_list.length/n; i++){
            for(int j=0; j<n;j++){
                answer[i][j]=num_list[a];
                a++;  
            }
        }
        return answer;
    }
}

 

+ 단일 반복문으로 풀이

i가 n의 배수가 되면 다른 차원의 배열로 넘어감으로 첫번째 배열 i/n

그리고 그 나머지가 2번째 배열의 갯수이므로 i%n로 표현 가능

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