아무 원소도 들어있지 않은 빈 배열 X가 있습니다.
길이가 같은 정수 배열 arr
과 boolean 배열 flag
가 매개변수로 주어질 때, flag
를 차례대로 순회하며 flag[i]
가 true라면 X의 뒤에 arr[i]
를 arr[i]
× 2 번 추가하고, flag[i]
가 false라면 X에서 마지막 arr[i]
개의 원소를 제거한 뒤 X를 return 하는 solution 함수를 작성해 주세요.
arr
의 길이 = flag
의 길이 ≤ 100arr
의 모든 원소는 1 이상 9 이하의 정수입니다.arr | flag | result |
---|---|---|
[3, 2, 4, 1, 3] | [true, false, true, false, false] | [3, 3, 3, 3, 4, 4, 4, 4] |
입출력 예 #1
i | flag[i] | arr[i] | X |
---|---|---|---|
[] | |||
0 | true | 3 | [3, 3, 3, 3, 3, 3] |
1 | false | 2 | [3, 3, 3, 3] |
2 | true | 4 | [3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 4, 4] |
3 | false | 1 | [3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 4] |
4 | false | 3 | [3, 3, 3, 3, 4, 4, 4, 4] |
import java.util.*;
class Solution {
public int[] solution(int[] arr, boolean[] flag) {
List<Integer> answer = new ArrayList<>();
for (int i = 0; i < flag.length; i++) {
if (flag[i]) {
for (int j = 0; j < arr[i] * 2; j++) {
answer.add(arr[i]);
}
} else {
for (int j = 0; j < arr[i]; j++) {
answer.remove(answer.size() - 1);
}
}
}
return answer.stream().mapToInt(i->i).toArray();
}
}