[백준] 12836번: 가계부 (Easy) - 파이썬

• 백준 링크: 12836번: 가계부 (Easy)
• solved.ac 난이도: 브론즈 II
• 시간 제한: 1 초
• 메모리 제한: 512 MB

문제

살아있는 화석이라고 불리는 월곡이는 돈에 찌들려 살아가고 있다. 그에게 있어 수입과 지출을 관리하는 것은 굉장히 중요한 문제이다. 스마트폰에 가계부 어플리케이션을 설치해서 사용하려 했지만, 월곡이는 굉장히 오래 살았기에 원하는 정보를 얻기에는 동작 속도가 너무나도 느렸다. 가끔 입력을 빼먹은 것이 생기면 다시 추가하고 계산하는 것도 느려서, 성격이 급한 월곡이는 결국 스마트폰을 부숴버리고 말았다. 월곡이를 도와주는 프로그램을 작성하기 위해, 아래와 같은 동작들을 처리하는 프로그램을 작성하시오.

입력

첫째 줄에 월곡이가 살아온 날 N, 쿼리의 개수 Q가 주어진다. (N ≤ 104, Q ≤ 103)

출력

각 2 쿼리에 대해 계산된 값을 각 줄에 출력한다.

코드

import sys; input = sys.stdin.readline

def solve():
    N, Q = map(int, input().split())
    money = [0] * (N + 1)

    for _ in range(Q):
        query, p, q = map(int, input().split())
        if query == 1:
            money[p] += q
        else:
            print(sum(money[p:q + 1]))

solve()