結果

問題 No.789 範囲の合計
ユーザー FuyuruFuyuru
提出日時 2024-02-16 18:21:40
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
TLE  
実行時間 -
コード長 1,601 bytes
コンパイル時間 370 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 76,624 KB
最終ジャッジ日時 2024-09-28 19:33:59
合計ジャッジ時間 3,078 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 28 ms
17,700 KB
testcase_01 AC 27 ms
10,752 KB
testcase_02 TLE -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
権限があれば一括ダウンロードができます

ソースコード

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