수열과 구간 쿼리 4
Programmers 코딩 기초 트레이닝
문제 설명
정수 배열 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 합니다.
class Solution {
public int[] solution(int[] arr, int[][] queries) {
for(int i = 0; i < queries.length; i++){
for(int j = queries[i][0]; j <= queries[i][1];j++){
if(j % queries[i][2] == 0)
arr[j]++;
}
}
return arr;
}
}
출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges
Leave a comment