[Programmers] 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합니다.

 

 

---

 

💡풀이

⭐제출 코드 (for문만 활용)

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

answer의 길이는 num_list 길이를 n으로 나눈 것이므로 num_list.length/n으로 설정해주고,

각 배열마다 n개씩 들어가므로 answer = new int[num_list.length/n][n]; 로 설정해준다. 

 

 

2중 for문을 활용하여answer[0][j]에 num_list의 0~n-1번째까지 n개 넣어주고, answer[1][j]에 num_list의 n~(n-1)번째까지 n개를 넣어주고,

 

이를 answer의 길이만큼 반복해주면 된다.

 

 

 

 

Arrays의 메소드를 활용하는 방법도 있다.

 

⭐Arrays.copyOf() 메소드 활용 코드

import java.util.Arrays;

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

위 코드는 Arrays의 copyOfRange() 메소드를 활용한 코드이다.

 

 

Arryas.copyOfRange 메소드는

Arrays.copyOfRange(복사하고자하는 배열, 시작 인덱스, 끝인덱스); 로 선언해주는데,

 

- 복사하고자하는 배열: num_list

- 시작 인덱스: idx      -> 0

- 끝 인덱스: idx+n    ->  n개씩 나눠주므로

 

 

for문을 실행하면

answer[0]에 인덱스 0~n-1까지 n개만큼 넣어주므로 answer[0][n]의 배열이 되고, idx는 n만큼 증가.

answer[1]에도 n~(n+n-1)까지 n개만큼 넣어주므로 answer[1][n]의 배열이 된다.

 

이를 answer의 길이만큼 반복해주고, answer을 return 해주면 된다.

 

 

 

 

자세한 copyOfRange() 메소드의 사용법은 따로 정리해 두었으니 아래 글을 참고하면 된다.

[Java] 배열 복사하기 Arrays.copyOf()

 

 

 

---

**실행시간 비교

왼) for문 활용  /  오) Arrays메소드 활용

위의 두 코드의 실행 시간을 비교해보면 for문을 활용했을 때의 시간이 더 짧다는 것을 확인할 수 있다.