정수 배열 arr
와 2차원 정수 배열 queries
이 주어집니다.
queries
의 원소는 각각 하나의 query
를 나타내며, [s, e, k]
꼴입니다.
각 query
마다 순서대로 s
≤ i
≤ e
인 모든 i
에 대해 i
가 k
의 배수이면 arr[i]
에 1을 더합니다.
위 규칙에 따라 queries
를 처리한 이후의 arr
를 return 하는 solution 함수를 완성해 주세요.
arr
의 길이 ≤ 1,000arr
의 원소 ≤ 1,000,000queries
의 길이 ≤ 1,000s
≤ e
< arr
의 길이k
≤ 5arr | queries | result |
---|---|---|
[0, 1, 2, 4, 3] | [[0, 4, 2],[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[] query : queries) {
int s = query[0];
int e = query[1];
int k = query[2];
for (int i = s; i < e + 1; i++) {
if (i % k == 0) {
arr[i] += 1;
}
}
}
return arr;
}
}