정수 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
의 길이 = 3slicer
에 담긴 정수를 차례대로 a, b, c라고 할 때num_list
의 길이 - 1num_list
의 길이 ≤ 30num_list
의 원소 ≤ 100n | slicer | num_list | result |
---|---|---|---|
3 | [1, 5, 2] | [1, 2, 3, 4, 5, 6, 7, 8, 9] | [2, 3, 4, 5, 6] |
4 | [1, 5, 2] | [1, 2, 3, 4, 5, 6, 7, 8, 9] | [2, 4, 6] |
입출력 예 #1
입출력 예 #2
import java.util.*;
class Solution {
public int[] solution(int n, int[] slicer, int[] num_list) {
List<Integer> answer = new ArrayList<>();
if (n == 1) {
for (int i = 0; i <= slicer[1]; i++) {
answer.add(num_list[i]);
}
} else if (n == 2) {
for (int i = slicer[0]; i < num_list.length; i++) {
answer.add(num_list[i]);
}
} else if (n == 3) {
for (int i = slicer[0]; i <= slicer[1]; i++) {
answer.add(num_list[i]);
}
} else {
for (int i = slicer[0]; i <= slicer[1]; i += slicer[2]) {
answer.add(num_list[i]);
}
}
return answer.stream().mapToInt(i->i).toArray();
}
}