[프로그래머스] 리스트 자르기 자바 java

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

프로그래머스

문제 설명
정수 n과 정수 3개가 담긴 리스트 slicer 그리고 정수 여러 개가 담긴 리스트 num_list가 주어집니다. slicer에 담긴 정수를 차례대로 a, b, c라고 할 때, n에 따라 다음과 같이 num_list를 슬라이싱 하려고 합니다.

n = 1 : num_list의 0번 인덱스부터 b번 인덱스까지
n = 2 : num_list의 a번 인덱스부터 마지막 인덱스까지
n = 3 : num_list의 a번 인덱스부터 b번 인덱스까지
n = 4 : num_list의 a번 인덱스부터 b번 인덱스까지 c 간격으로
올바르게 슬라이싱 한 리스트를 return 하도록 solution 함수를 완성해 주세요.

제한사항
n 은 1, 2, 3, 4 중 하나입니다.
slicer의 길이 = 3
slicer에 담긴 정수를 차례대로 a, b, c라고 할 때

0 ≤ a ≤ b ≤ num_list의 길이 - 1
1 ≤ c ≤ 3
5 ≤ num_list의 길이 ≤ 30

0 ≤ num_list의 원소 ≤ 100

---

입력받은 정수 n에 따라 배열 slice로부터 받은 인덱스를 사용하여 num_list의 일부를 반환하는 코드를 작성하라는 문제이다. 

num_list의 일부 배열을 가져오기 위해서 필요한 것은 자르는데 필요한 시작인덱스와 마지막 인덱스, 간격 그리고 num_list배열이다. 

위 요소를 가지고 배열을 잘라줄 함수를 호출하여 배열을 잘라 반환하도록 하면 된다. 

 

 

전체코드

import java.util.*;
class Solution {
    public int[] solution(int n, int[] slicer, int[] num_list) {
        ArrayList<Integer> arr = new ArrayList<>();
        
        int a = slicer[0];
        int b = slicer[1];
        int c = slicer[2];
        
        switch(n){
            case 1: 
                arr= getlist(0,b,1,num_list);
                break;
            case 2: 
                arr= getlist(a,num_list.length-1,1,num_list);
                break;
            case 3: 
                arr= getlist(a,b,1,num_list);
                break;
            case 4: 
                arr= getlist(a,b,c,num_list);
                break;
                
        }
        int[] answer =new int[arr.size()];
        for(int i=0; i<arr.size();i++)
            answer[i]=arr.get(i);
        return answer;
    }
    
    private ArrayList<Integer> getlist(int start, int end, int step,int[] num_list){
        ArrayList<Integer> arr = new ArrayList<>();
        
        for(int i=start; i<=end;i+=step){
            arr.add(num_list[i]);
        }
        
        return arr;
    }
}