[programmers] 수열과 구간 쿼리 4 - JavaScript

정수 배열 arr와 2차원 정수 배열 queries가 주어지고, queries에 따라 arr를 처리한 후의 결과를 반환하는 solution 함수를 작성해야 합니다. 각 query는 [s, e, k] 형태이며, s ≤ i ≤ e인 모든 i에 대해 i가 k의 배수이면 arr[i]에 1을 더합니다. 문제를 풀면서 문제 이해에 어려움을 겪었지만, 기존 수열과 구간 쿼리 문제와 유사한 방식으로 if문을 사용하여 해결할 수 있습니다.
Jan 06, 2024
[programmers] 수열과 구간 쿼리 4 - JavaScript

문제 설명

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

제한사항

  • 1 ≤ arr의 길이 ≤ 1,000
    • 0 ≤ arr의 원소 ≤ 1,000,000
  • 1 ≤ queries의 길이 ≤ 1,000
    • 0 ≤ s ≤ e < arr의 길이
    • 0 ≤ k ≤ 5

입출력 예

arr
queries
result
[0, 1, 2, 4, 3]
[[0, 4, 1],[0, 3, 2],[0, 3, 3]]
[3, 2, 4, 6, 4]

입출력 예 설명

입출력 예 #1
  • 각 쿼리에 따라 arr가 다음과 같이 변합니다.
arr
[0, 1, 2, 4, 3]
[1, 2, 3, 5, 4]
[2, 2, 4, 5, 4]
[3, 2, 4, 6, 4]
  • 따라서 [3, 2, 4, 6, 4]를 return 합니다.

※ 2023년 04월 27일 입출력 예 설명이 수정되었습니다.

solution.js

function solution(arr, queries) { for(let i = 0 ; i < queries.length ; i ++){ const [s,e,k] = queries[i]; for(let j = 0 ; j <arr.length; j++){ if(s<=j && j<=e){ if(j % k == 0){ arr[j] += 1; } } } } return arr; }
 

핵심 키워드

  • 기존 수열과 구간 쿼리 문제와 같이 if문을 통해 풀 수 있다.
  • 문제에서 정의하는 i는 arr[i]가 아닌 index의 i다.
 

결론!

해당 문제를 풀면서 문제를 이해하는 것에 어려움을 느꼈다. 천천히 문제를 구성하는 방식의 중요함을 느낄 수 있었다.
 
Share article

More articles

See more posts

👨🏻‍💻DriedPollack's Blog