結果
問題 | No.789 範囲の合計 |
ユーザー |
|
提出日時 | 2022-06-25 18:34:55 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 331 ms / 1,000 ms |
コード長 | 897 bytes |
コンパイル時間 | 374 ms |
コンパイル使用メモリ | 82,200 KB |
実行使用メモリ | 113,444 KB |
最終ジャッジ日時 | 2024-11-15 05:57:39 |
合計ジャッジ時間 | 5,621 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 15 |
ソースコード
import bisect class Bit: def __init__(self, n): self.size = n self.tree = [0] * (n + 1) def sum(self, i): s = 0 while i > 0: s += self.tree[i] i -= i & -i return s def add(self, i, x): while i <= self.size: self.tree[i] += x i += i & -i n = int(input()) qry = [] idx = [] for i in range(n): q = [int(i) for i in input().split()] qry.append(q) if q[0] == 0: idx.append(q[1]) asrt = sorted(set(idx)) adt = { v: i for i, v in enumerate(asrt) } ki = Bit(len(adt)+1) ans = 0 #print(bisect.bisect_left(asrt,5)+1) #print(asrt) for i in range(n): q,x,y = qry[i] if q == 0: ki.add(adt[x]+1,y) else: l = bisect.bisect_left(asrt,x) r = bisect.bisect_right(asrt,y) ans += ki.sum(r) - ki.sum(l) #print(ans,l,r) print(ans)