프로그래머스 - 2차원으로 만들기 / JAVA (자바) 코딩 알고리즘 연습

반응형

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