코딩테스트/코테_java

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

숭늉다섯 2023. 12. 6. 19:50
문제
정수 리스트 num_list와 정수 n이 주어질 때, n 번째 원소부터 마지막 원소까지의 모든 원소를 담은 리스트를 return하도록 solution 함수를 완성해주세요.

 

풀이
배열에서 n 번째 만큼 뺀 후, 0부터 시작이기 때문에 +1을 해준다. 
그리고 n 번째 부터 반환
class Solution {
    public int[] solution(int[] num_list, int n) {
        int[] answer = new int[num_list.length-n+1];
            for(int i=0; i< num_list.length-n+1;i++){
                answer[i]= num_list[i+n-1];
            }
        return answer;
    }
}

 

다른사람 풀이
copyOfRange 함수라는 것을 사용
: Arrays.copyOfRange(복사할 배열, 복사 시작 인덱스, 복사 끝 인덱스); 
import java.util.*;
class Solution {
    public int[] solution(int[] num_list, int n) {
        int[] a= Arrays.copyOfRange(num_list, n-1, num_list.length);
        return a;
    }
}