백준/ Gold 1 문제 , 백준 파이썬 2268 , 수들의 합 7 [세그먼트 트리]

백준/ Gold 1 문제 , 백준 파이썬 2268 , 수들의 합 7 [세그먼트 트리]

 

 

Check Point !   ( 해당사항 ✓체크 )

1. 막힘 없이 수월하게 풀린 문제인가?

2. 1시간이내로 풀렸던 문제인가?

3. 1시간 이상 or 며칠을 두고 풀어봤더니 풀린 문제인가?

4. 시간을 써도 도무지 풀 수 없는 문제인가?

5. 솔루션을 찾아봤는가?

-------------------------------------------------------------------------------------------
난이도 체감
1. 최상
2. 상
3. 중
4. 하


<이해도>
1. 완벽히 이해
2. 다소 헷갈리는 부분들이 있음
3. 이해 못함


<덧붙일 말>
python3 로는 계속 시간 초과발생, pypy3 로 통과
무조건 i가 j보다 작은 것이 아니니 주의. 두 경우로 나눠줘야됨.

 

 

 

문제 출처

https://www.acmicpc.net/problem/2268

2268번: 수들의 합 7

 

 

 

 

 

정답

 

pypy3로 통과

 

 

import sys
input=sys.stdin.readline

def init(start,end,index):
    if start==end:
        tree[index]=arr[start]
    else:
        mid=(start+end)//2
        tree[index]=init(start,mid,index*2)+init(mid+1,end,index*2+1)
    return tree[index]

def update(start,end,index,w,v):
    if w<start or w>end:
        return
    if start==end:
        tree[index]=v
        return tree[index]
    mid=(start+end)//2
    update(start,mid,index*2,w,v)
    update(mid+1,end,index*2+1,w,v)
    tree[index]=tree[index*2]+tree[index*2+1]
    
def range_sum(start,end,index,left,right):
    if left>end or right<start:
        return 0
    if start>=left and right>=end:
        return tree[index]
    mid=(start+end)//2
    return range_sum(start,mid,index*2,left,right)+range_sum(mid+1,end,index*2+1,left,right)
    
N,M=map(int,input().split())
arr=[ 0 for _ in range(N)]
tree=[0]*(N*4)

init(0,N-1,1)

for i in range(M):
    a,b,c=map(int,input().split())
    if a==0:
        if b>c:
            print(range_sum(0,N-1,1,c-1,b-1))
        else:
            print(range_sum(0,N-1,1,b-1,c-1))
    elif a==1:
        arr[b-1]=c
        update(0,N-1,1,b-1,c)