[프로그래머스 코딩테스트] 2차원으로 만들기(Java)

  문제 설명

  정수 배열 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]]

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

  우선 answer 배열의 크기를 설정합니다. 행열을 각각 num_list의 길이를 n으로 나눈 값과 n으로 설정합니다.

  행이 될 예정인, 바깥 for문이 num_list의 길이를 n으로 나눈 값만큼 순회합니다. 열이 될 예정인, 안쪽 for문이 n만큼 순회합니다. answer[i][j]는 인덱싱을 해주려는 것입니다. num_list에서 할당하려는 1차원 배열에서의 원소 위치입니다. 첫번째 입출력 예를 예시로 들어보겠습니다.

 

answer[i][j] = num_list[i * n + j] answer[0][0] = num_list[0] answer[0][1] = num_list[1] answer[1][0] = num_list[2] answer[1][1] = num_list[3] answer[1][2] = num_list[4] answer[2][0] = num_list[5] answer[2][1] = num_list[6] answer[2][2] = num_list[7]

 

  for문을 풀이해보면 다음과 같은 식이 나옵니다. 바깥 for문이 한 번 순회하는 동안에 안쪽 for문은 여러 번 순회합니다.

  다른 사람의 풀이 중 바깥 for문만을 이용한 뒤 [i][j] 부분을 n과 i의 계산으로 채우는 풀이도 있습니다만, 거기까지는 미처 생각을 못했네요.