[프로그래머스] 코딩테스트 연습 - n번째 원소까지(JAVA)

https://school.programmers.co.kr/learn/courses/30/lessons/181889

프로그래머스

 

문제

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

---

소스코드

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;
    }
}

 

소스코드 설명

1. int형 배열을 n의 크기로 선언

2. 0부터 n까지 num_list을 복사

---

Arrays.copyOfRange를 사용하면 더 간단하게 구현이 가능하다.

import java.util.*;

class Solution {
    public int[] solution(int[] num_list, int n) {
        int[] answer = {};

        answer = Arrays.copyOfRange(num_list,0,n);

        return answer;
    }
}

copyOfRange(원본배열, 복사할 첫 인덱스, 복사할 마지막 인덱스)

위와 같이 범위를 지정해서 사용 가능하다.