[자바] n번째 원소까지

https://school.programmers.co.kr/learn/courses/30/lessons/181889?language=java 

프로그래머스

 

문제 설명

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

 

제한사항

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

입출력 예 설명

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

 

나의 문제 풀이

1) return 할 answer의 배열의 크기를 설정해 준다.

2) for문을 돌면서 answer에 num_list값을 집어 넣는다.

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