[프로그래머스] JAVA 0단계 n번째 원소까지

n번째 원소까지

정수 리스트 num_list와 정수 n이 주어질 때, n 번째 원소부터 마지막 원소까지의 모든 원소를 담은 리스트를 return하도록 solution 함수를 완성해주세요.

 

풀이

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

 

결과

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

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