[Programmers] 리스트 자르기 (JAVA, PYTHON)

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

프로그래머스

Algorithm

---

Python

def solution(n, slicer, num_list):
    a,b,c=slicer
    if n==1: return num_list[:b+1]
    if n==2: return num_list[a:]
    if n==3: return num_list[a:b+1]
    return num_list[a:b+1:c]

Java

import java.util.*;

class Solution {
    public int[] solution(int n, int[] slicer, int[] num_list) {
        List<Integer> li = new ArrayList<>();
        
        int a = slicer[0], b = slicer[1], c = slicer[2];
        
        if(n == 1){
            for(int i = 0; i <=b ; i++){
                li.add(num_list[i]);
            }
        }
        else if(n == 2){
            for(int i = a; i <num_list.length ; i++){
                li.add(num_list[i]);
            }
        }
        else if(n == 3){
            for(int i = a; i <=b ; i++){
                li.add(num_list[i]);
            }
        }
        else if(n == 4){
            for(int i = a; i <=b ; i+=c){
                li.add(num_list[i]);
                ;
            }
        }
        return li.stream().mapToInt(x -> x).toArray();
    }
}