[프로그래머스] JAVA 0단계 n개 간격의 원소들

n개 간격의 원소들

정수 리스트 num_list와 정수 n이 주어질 때, num_list의 첫 번째 원소부터 마지막 원소까지 n개 간격으로 저장되어있는 원소들을 차례로 담은 리스트를 return하도록 solution 함수를 완성해주세요.

 

풀이

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

 

결과

테스트 1
입력값 〉	[4, 2, 6, 1, 7, 6], 2
기댓값 〉	[4, 6, 7]
실행 결과 〉	테스트를 통과하였습니다.

테스트 2
입력값 〉	[4, 2, 6, 1, 7, 6], 4
기댓값 〉	[4, 7]
실행 결과 〉	테스트를 통과하였습니다.