정수 배열 arr
가 주어집니다. 배열 안의 2가 모두 포함된 가장 작은 연속된 부분 배열을 return 하는 solution 함수를 완성해 주세요.
단, arr
에 2가 없는 경우 [-1]을 return 합니다.
arr
의 길이 ≤ 100,000arr
의 원소 ≤ 10arr | result |
---|---|
[1, 2, 1, 4, 5, 2, 9] | [2, 1, 4, 5, 2] |
[1, 2, 1] | [2] |
[1, 1, 1] | [-1] |
[1, 2, 1, 2, 1, 10, 2, 1] | [2, 1, 2, 1, 10, 2] |
입출력 예 #1
입출력 예 #2
입출력 예 #3
입출력 예 #4
import java.util.*;
class Solution {
public int[] solution(int[] arr) {
List<Integer> count = new ArrayList<>();
for (int i = 0; i < arr.length; i++) {
if (arr[i] == 2) {
count.add(i);
}
}
List<Integer> answer = new ArrayList<>();
if (!count.isEmpty()) {
for (int i = count.get(0); i <= count.get(count.size() - 1); i++) {
answer.add(arr[i]);
}
} else {
answer.add(-1);
}
return answer.stream().mapToInt(i->i).toArray();
}
}