프로그래머스 리스트 자르기 (JAVA)

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

프로그래머스

 

저는 구현을 통해 문제를 해결했습니다.

start end step으로 각각의 n에 따라 이 값을 다르게 초기화하고 for문을 통해 구현했습니다.

 

import java.util.*;
class Solution {
    public int[] solution(int n, int[] slicer, int[] num_list) {
        int[] answer = split(n, slicer, num_list);
        return answer;
    }
    private static int[] split(int n, int[] slicer, int[] num_list){
        int start = 0;
        int end = num_list.length-1;
        int step = 1;
        if(n == 1){
            end = slicer[1];
        }
        if(n == 2){
            start = slicer[0];
        }
        if(n == 3){
            start = slicer[0];
            end = slicer[1];
        }
        if(n == 4){
            start = slicer[0];
            end = slicer[1];
            step = slicer[2];
        }
        int size = (end - start) / step + 1;
        int[] result = new int[size];
        int index = 0;
        for(int i = start ; i <= end ; i += step){
            result[index++] = num_list[i];
        }
        return result;
    }
}