아무 원소도 들어있지 않은 빈 배열 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
예제 1번에서 X의 변화를 표로 나타내면 다음과 같습니다
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] |
따라서 [3, 3, 3, 3, 4, 4, 4, 4]를 return 합니다.
function solution(arr, flag) {
return arr.reduce((a,c,i)=> flag[i] ? [...a, ...Array(c*2).fill(c)] : a.slice(0,-c),[])
}
다른풀이
function solution(arr, flag) {
return arr.reduce((prev, num, i) => (flag[i] ? [...prev, ...new Array(num * 2).fill(num)] : prev.slice(0, -num)),[],);
}