結果

問題 No.789 範囲の合計
ユーザー FuyuruFuyuru
提出日時 2024-02-16 18:20:56
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 1,601 bytes
コンパイル時間 716 ms
コンパイル使用メモリ 82,404 KB
実行使用メモリ 196,156 KB
最終ジャッジ日時 2024-09-28 19:33:56
合計ジャッジ時間 7,613 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 35 ms
52,456 KB
testcase_01 AC 34 ms
52,600 KB
testcase_02 MLE -
testcase_03 AC 226 ms
76,588 KB
testcase_04 MLE -
testcase_05 MLE -
testcase_06 MLE -
testcase_07 AC 200 ms
76,324 KB
testcase_08 MLE -
testcase_09 MLE -
testcase_10 MLE -
testcase_11 MLE -
testcase_12 MLE -
testcase_13 AC 34 ms
52,192 KB
testcase_14 AC 33 ms
52,596 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
#print = sys.stdout.write
class SegTree:
    def __init__(self,N,op,identity):
        self.op = op
        self.identity = identity
        self.N = N
        self.data = dict()
    def __getitem__(self,key):
        key += self.N
        if key in self.data:
            return self.data[key]
        else:
            return self.identity
    def set_value(self,x,val):
        x += self.N
        self.data[x] = val
        while x > 1:
            x >>= 1
            if x<<1|0 in self.data:
                l = self.data[x<<1|0]
            else:
                l = self.identity
            if x<<1|1 in self.data:
                r = self.data[x<<1|1]
            else:
                r = self.identity
            self.data[x] = self.op(l,r)
    def fold(self,l,r):
        res = self.identity
        l,r = l+self.N,r+self.N
        while l < r:
            if l&1:
                if l in self.data:
                    lv = self.data[l]
                else:
                    lv = self.identity
                res = self.op(res,lv)
                l += 1
            if r&1:
                r -= 1
                if r in self.data:
                    rv = self.data[r]
                else:
                    rv = self.identity
                res = self.op(res,rv)
            l,r = l>>1,r>>1
        return res


seg = SegTree(1000000001,lambda x,y:x+y,0)
Q = int(input())
ans = 0
for _ in range(Q):
    op,x,y = map(int,input().split())
    if op == 0:
        seg.set_value(x,seg[x]+y)
    else:
        ans += seg.fold(x,y+1)
print(ans)
0