💻 자바 알고리즘 - n 번째 원소까지

📕 문제 설명

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

 

📕 제한 사항

• 2 ≤ num_list의 길이 ≤ 30
• 1 ≤ num_list의 원소 ≤ 9
• 1 ≤ n ≤ 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;
    }
}

정수형 배열 num_list 원소중에서 n개만 담은 배열을 리턴하는 문제.

먼저 n개 담을 배열 answer를 만들어주고 for문을 돌면서 num_list의 원소를 answer에 담는다.

 

import java.util.Arrays;

class Solution {
    public int[] solution(int[] num_list, int n) {
        return Arrays.copyOfRange(num_list, 0, n);
    }
}

for문을 쓰지 않고 Arrays.copyOfRange()를 통해 배열을 통채로 카피해서 return 하는 방법도 있다.

 

💡 참고

• Arrays.copyOfRange(배열,첫번째인덱스,마지막인덱스) : 인덱스를 지정하여 배열을 카피하는 함수