빈 배열에 추가, 삭제하기
📝 문제
아무 원소도 들어있지 않은 빈 배열 X가 있습니다. 길이가 같은 정수 배열 arr과 boolean 배열 flag가 매개변수로 주어질 때, flag를 차례대로 순회하며 flag[i]가 true라면 X의 뒤에 arr[i]를 arr[i] × 2 번 추가하고, flag[i]가 false라면 X에서 마지막 arr[i]개의 원소를 제거한 뒤 X를 return 하는 solution 함수를 작성해 주세요.
⛔️ 제한 사항
- 1 ≤ arr의 길이 = flag의 길이 ≤ 100
- arr의 모든 원소는 1 이상 9 이하의 정수입니다.
- 현재 X의 길이보다 더 많은 원소를 빼는 입력은 주어지지 않습니다.
💡 풀이
function solution(arr, flag) {
var X = [];
for(let i=0; i<flag.length; i++){
if(flag[i]){
for(let j=0; j<arr[i]*2; j++){
X.push(arr[i]);
}
}else{
for(let j=0; j<arr[i]; j++){
X.pop();
}
}
}
return X;
}=> flag[i] === true면 X배열에 arr[i]를 arr[i] × 2 번만큼 추가,
flag[i] === false면 X배열 마지막에서 arr[i]개의 원소를 제거.
반복문을 반복해서 사용하면서 이렇게 푸는게 맞나 싶었다.
다른 간결한 방법이 있을 것 같다.
✅ 다른 풀이
function solution(arr, flag) {
let X = [];
flag.forEach((el, index) => {
if(el) {
for(let i = 0; i < arr[index] * 2; i++) {
X.push(arr[index])
}
} else if (!el){
for(let i = 0; i < arr[index]; i++){
X.pop()
}
}
})
return X
}=> 내 풀이와 같은 방법이나 for문 대신 forEach를 사용했다.
answer = answer.splice(0,answer.length-arr[i])=> for문과 arr.pop()을 사용하는 대신 splice를 이용했다.
function solution(arr, flag) {
return arr.reduce(
(prev, num, i) => (flag[i] ? [...prev, ...new Array(num * 2).fill(num)] : prev.slice(0, -num)),
[],
);
}=> 코드가 간결하지만 처리속도가 빠른 것은 아니다.
🚀 TIL
🚀 Front-end Dev