[JavaScript] 프로그래머스 빈 배열에 추가, 삭제하기

프로그래머스

문제 설명

아무 원소도 들어있지 않은 빈 배열 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	result
[3, 2, 4, 1, 3]	[true, false, true, false, false]	[3, 3, 3, 3, 4, 4, 4, 4]

코드

const solution = (arr, flag) => {
    let X = [];
    
    for (let i = 0; i < arr.length; i++) {
        if (flag[i]) {
            for (let j = 0; j < arr[i] * 2; j++) {
                X.push(arr[i]);
            }
        } else {
            X.splice(X.length - arr[i], arr[i]);
        }
    }
    
    return X;
}

 

설명

flag[i]가 true인 경우에는 arr[i]의 값을 2배한 횟수만큼 x에 추가하고 flag[i]가 false인 경우에는 x의 마지막 arr[i]개의 요소를 제거합니다. 반복문이 끝난 후 x을 반환합니다.