[프로그래머스] n 번째 원소부터

sungeun·2023년 6월 12일
0

코딩 기초 트레이닝

목록 보기
61/124
post-thumbnail

📔 문제 설명

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

📌 제한사항

  • 2 ≤ num_list의 길이 ≤ 30
  • 1 ≤ num_list의 원소 ≤ 9
  • 1 ≤ nnum_list의 길이

🔍 입출력 예

num_listnresult
[2, 1, 6]3[6]
[5, 2, 1, 7, 5]2[2, 1, 7, 5]

🔍 입출력 예 설명

입출력 예 #1

  • [2, 1, 6]의 세 번째 원소부터 마지막 원소까지의 모든 원소는 [6]입니다.

입출력 예 #2

  • [5, 2, 1, 7, 5]의 두 번째 원소부터 마지막 원소까지의 모든 원소는 [2, 1, 7, 5]입니다.

🐹 내가 작성한 코드

import java.util.*;

class Solution {
    public int[] solution(int[] num_list, int n) {
        List<Integer> answer = new ArrayList<>();
        
        for (int i = n; i <= num_list.length; i++) {
            answer.add(num_list[i-1]);
        }
        
        return answer.stream().mapToInt(i->i).toArray();
    }
}
profile
Back-end Developer

0개의 댓글