프로그래머스 LV.0 빈 배열에 추가, 삭제하기

프로그래머스 LV.0 빈 배열에 추가, 삭제하기

https://school.programmers.co.kr/learn/courses/30/lessons/181860

프로그래머스

 

1. 문제 명

빈 배열에 추가, 삭제하기

2. 문제 설명

아무 원소도 들어있지 않은 빈 배열 X가 있습니다. 길이가 같은 정수 배열 arr과 boolean 배열 flag가 매개변수로 주어질 때, flag를 차례대로 순회하며 flag[i]가 true라면 X의 뒤에 arr[i]를 arr[i] × 2 번 추가하고, flag[i]가 false라면 X에서 마지막 arr[i]개의 원소를 제거한 뒤 X를 return 하는 solution 함수를 작성해 주세요.

3. 제한 사항

- 1 ≤ arr의 길이 = flag의 길이 ≤ 100

- arr의 모든 원소는 1 이상 9 이하의 정수입니다.

- 현재 X의 길이보다 더 많은 원소를 빼는 입력은 주어지지 않습니다.

4. 예시

arr	flag	result
[3, 2, 4, 1, 3]	[true, false, true, false, false]	[3, 3, 3, 3, 4, 4, 4, 4]

5. 기본 제공 코드

function solution(arr, flag) {
    var answer = [];
    return answer;
}

6. 제출한 내 답

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

 

6-2. VSC에 작성한 내용

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

// 테스트
console.log(solution([3, 2, 4, 1, 3], [true, false, true, false, false]));

7. 특이사항

그냥 reduce map filter 욕심내지말고 for문이나 while문을 풀어야겠다

일단 푸는게 더 중요하니까

8. 다른 사람이 작성한 답

8-1. 가장 많이 쓰인 풀이법

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
}

나와 같음

 

8-2. reduce

function solution(arr, flag) {
  return arr.reduce(
    (prev, num, i) => (flag[i] ? [...prev, ...new Array(num * 2).fill(num)] : prev.slice(0, -num)),
    [],
  );
}