結果
問題 |
No.789 範囲の合計
|
ユーザー |
|
提出日時 | 2025-07-03 22:50:25 |
言語 | PyPy3 (7.3.15) |
結果 |
MLE
|
実行時間 | - |
コード長 | 1,760 bytes |
コンパイル時間 | 595 ms |
コンパイル使用メモリ | 82,372 KB |
実行使用メモリ | 149,920 KB |
最終ジャッジ日時 | 2025-07-03 22:50:31 |
合計ジャッジ時間 | 5,781 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 8 MLE * 7 |
ソースコード
N = int(input()) class SegmentTree: # 初期化処理 # f : SegmentTreeにのせるモノイド # default : fに対する単位元 def __init__(self, size, f=lambda x,y : x+y, default=0): self.N = size self.size = 2**(size-1).bit_length() # 簡単のため要素数Nを2冪にする self.default = default self.dat = [default]*(self.size*2) # 要素を単位元で初期化 self.f = f def update(self, i, x): i += self.size self.dat[i] = x while i > 0: i >>= 1 self.dat[i] = self.f(self.dat[i*2], self.dat[i*2+1]) def query(self, l, r): l += self.size r += self.size lres, rres = self.default, self.default while l < r: if l & 1: lres = self.f(lres, self.dat[l]) l += 1 if r & 1: r -= 1 rres = self.f(self.dat[r], rres) # モノイドでは可換律は保証されていないので演算の方向に注意 l >>= 1 r >>= 1 res = self.f(lres, rres) return res def __str__(self): return ' '.join([str(self.query(i, i+1)) for i in range(self.N)]) q = [0] * N for i in range(N): q[i] = list(map(int, input().split())) s = set() for i in range(N): if q[i][0] == 0: s.add(q[i][1]) else: s.add(q[i][1]) s.add(q[i][2]) s = sorted(s) l = len(s) d = dict() for i in range(len(s)): d[s[i]] = i s = SegmentTree(l) ans = 0 for i in range(N): if q[i][0] == 0: x = d[q[i][1]] s.update(x, s.query(x, x+1) + q[i][2]) else: l = d[q[i][1]] r = d[q[i][2]] ans += s.query(l, r+1) print(ans)