<Programmers> 수열과 구간 쿼리4

·2024년 12월 8일
1

프로그래머스

목록 보기
14/18
post-thumbnail

문제

정수 배열 arr와 2차원 정수 배열 queries이 주어집니다. queries의 원소는 각각 하나의 query를 나타내며, [s, e, k] 꼴입니다.
query마다 순서대로 s ≤ i ≤ e인 모든 i에 대해 ik의 배수이면 arr[i]에 1을 더합니다.
위 규칙에 따라 queries를 처리한 이후의 arrreturn 하는 solution 함수를 완성해 주세요.


문제 접근

첫 번째 query 범위 예시 :
0 =< index <= 4 일 때, 첫 번째 매개변수인 arr[i]에 1을 더해야함


문제 풀이

function solution(arr, queries) {
  for (const [s, e, k] of queries) {
    for (let i = s; i <= e; i++) {
      if (i % k === 0) {
        // 인덱스랑 k랑 나눴을 때, 0이여야 인덱스가 k의 배수임
        arr[i] += 1;
      }
    }
  }
  console.log(arr);
  return arr;
}

solution(
  [0, 1, 2, 4, 3],
  [
    [0, 4, 1],
    [0, 3, 2],
    [0, 3, 3],
  ]
); // [3, 2, 4, 6, 4]

풀이 핵심 포인트

  for (const [s, e, k] of queries) {...}

배열 구조 분해 할당으로 편하게 풀이 가능.


다른 풀이

  1. reduce 메서드를 활용한 다른 풀이
function solution(arr, queries) {
    return queries.reduce((bucket, [s, e, k]) => {
        for (let i = s; i <= e; i += 1) {
            if (i % k === 0) bucket[i] += 1
        }
        return bucket
    }, [...arr])
}
profile
- 배움에는 끝이 없다.

0개의 댓글