수열과 구간 쿼리 1

송용진·2023년 8월 28일
0

알고리즘과 자료구조

목록 보기
90/190

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

각 query마다 순서대로
s ≤ i ≤ e인 모든 i에 대해
arr[i]에 1을 더합니다.

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

내 코드

def solution(arr, queries):
    answer = []
    for j in range(len(queries)):
        for i in range(queries[j][0],queries[j][1]+1):
            arr[i] += 1
    answer = arr
    return answer

예시 코드1

def solution(arr, queries):
    for (s, e) in queries:
        arr = [a+1 if s <= i <= e else a for i, a in enumerate(arr)]
    return arr

예시 코드2

import numpy as np

def solution(arr, queries):
    answer = []

    arr = np.array(arr)
    for query in queries:
        arr[query[0]:query[1] + 1] += 1

    answer = arr.tolist()
    return answer
profile
개발자

0개의 댓글