Algorithm/프로그래머스

[알고리즘] 리스트 자르기 Java

rw- 2023. 5. 7. 17:49
728x90

import java.util.ArrayList;
import java.util.List;

class Solution {
    public int[] solution(int n, int[] slicer, int[] num_list) {
        List<Integer> list = new ArrayList<>();
        switch (n) {
          case 1:
            for (int i = 0; i <= slicer[1]; i++) {
              list.add(num_list[i]);
            }
            break;
          case 2:
            for (int i = slicer[0]; i < num_list.length; i++) {
              list.add(num_list[i]);
            }
            break;
          case 3:
            for (int i = slicer[0]; i <= slicer[1]; i++) {
              list.add(num_list[i]);
            }
            break;
          case 4:
            for (int i = slicer[0]; i <= slicer[1]; i+=slicer[2]) {
              list.add(num_list[i]);
            }
            break;
        }

        return list.stream().mapToInt(Integer::intValue).toArray();
    }
}

 

728x90
반응형