[프로그래머스] 수열과 구간 쿼리 4 (Java, Python)

문제

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

프로그래머스

 

코드(Java)

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];
			
			// k로 나누어 떨어지는 시작 인덱스 찾기
			int start = (s % k == 0) ? s : s + (k - s % k);
			
			for (int i = start; i <= e; i += k) {
				arr[i]++;
			}
		}
        return arr;
    }
}

 

코드(Python)

def solution(arr, queries):
	for s, e, k in queries:
		# k로 나누어 떨어지는 시작 인덱스 찾기
		start = (s + (k - s % k)) if s %k != 0 else s
		for i in range(start, e + 1, k):
			arr[i] += 1
	return arr