[프로그래머스] n 번째 원소까지

sungeun·2023년 6월 12일
0

코딩 기초 트레이닝

목록 보기
64/124
post-thumbnail

📔 문제 설명

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

📌 제한사항

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

🔍 입출력 예

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

🔍 입출력 예 설명

입출력 예 #1

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

입출력 예 #2

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

🐹 내가 작성한 코드

import java.util.*;

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

0개의 댓글