수열과 구간 쿼리 4

송용진·2023년 9월 6일
0

알고리즘

목록 보기
108/173

정수 배열 arr와
2차원 정수 배열 queries이 주어집니다.
queries의 원소는 각각 하나의 query를 나타내며,
[s, e, k] 꼴입니다.

각 query마다 순서대로
s ≤ i ≤ e인 모든 i에 대해
i가 k의 배수이면
arr[i]에 1을 더합니다.

위 규칙에 따라 queries를 처리한 이후의
arr를 return 하는 solution 함수를 완성해 주세요.

내 코드

def solution(arr, queries):
    answer = []
    for q in queries:
        s,e,k = q[0],q[1],q[2]
        for i in range(s,e+1):
            if i % k == 0:
                arr[i] += 1        
    answer = arr
    return answer

예시 코드

def solution(arr, queries):
    for s, e, k in queries:
        for i in range(s, e+1):
            if i%k == 0:
                arr[i] += 1
    return arr
profile
백엔드 개발자

0개의 댓글