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

문제출처

프로그래머스

---

문제

 

입출력 예

 

소스코드

import java.util.*;
class Solution {
    public List<Integer> solution(int n, int[] slicer, int[] num_list) {
        List<Integer> list = new ArrayList<>();
        
        int a = slicer[0];
        int b = slicer[1];
        int c = slicer[2];
        
        if(n==1){
            for(int i=0; i<=b; i++){
                list.add(num_list[i]);
            }
        }else if(n==2){
            for(int i=a; i<=num_list.length-1; i++){
                list.add(num_list[i]);
            }
        }else if(n==3){
            for(int i=a; i<=b; i++){
                list.add(num_list[i]);
            }
        }else{
            for(int i=a; i<=b; i+=c){
                list.add(num_list[i]);
            }
        }
        return list;
    }
}

 

풀이

slicer의 a, b, c를 구별하기 쉽게 변수에 담고, 조건문 안에 반복문을 통해 선언한 list에 적절한 값을 담아 return해준다.