結果
問題 | No.789 範囲の合計 |
ユーザー | 👑 rin204 |
提出日時 | 2022-09-30 23:00:10 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 895 ms / 1,000 ms |
コード長 | 1,286 bytes |
コンパイル時間 | 165 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 50,108 KB |
最終ジャッジ日時 | 2024-06-02 06:29:10 |
合計ジャッジ時間 | 9,105 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 27 ms
10,752 KB |
testcase_01 | AC | 27 ms
10,880 KB |
testcase_02 | AC | 877 ms
43,268 KB |
testcase_03 | AC | 498 ms
19,072 KB |
testcase_04 | AC | 808 ms
42,920 KB |
testcase_05 | AC | 849 ms
42,820 KB |
testcase_06 | AC | 812 ms
42,876 KB |
testcase_07 | AC | 479 ms
19,584 KB |
testcase_08 | AC | 662 ms
31,984 KB |
testcase_09 | AC | 620 ms
31,284 KB |
testcase_10 | AC | 895 ms
50,108 KB |
testcase_11 | AC | 786 ms
42,528 KB |
testcase_12 | AC | 749 ms
42,584 KB |
testcase_13 | AC | 27 ms
10,752 KB |
testcase_14 | AC | 26 ms
10,752 KB |
ソースコード
class Bit: def __init__(self, n): self.size = n self.n0 = 1 << (n.bit_length() - 1) self.tree = [0] * (n + 1) def range_sum(self, l, r): return self.sum(r - 1) - self.sum(l - 1) def sum(self, i): i += 1 s = 0 while i > 0: s += self.tree[i] i -= i & -i return s def get(self, i): return self.sum(i) - self.sum(i - 1) def add(self, i, x): i += 1 while i <= self.size: self.tree[i] += x i += i & -i def lower_bound(self, x): pos = 0 plus = self.n0 while plus > 0: if pos + plus <= self.size and self.tree[pos + plus] < x: x -= self.tree[pos + plus] pos += plus plus //= 2 return pos n = int(input()) query = [] X = set() for _ in range(n): t, x, y = map(int, input().split()) X.add(x) if t == 1: y += 1 X.add(y) query.append((t, x, y)) n = len(X) lst = sorted(X) dic = {l:i for i, l in enumerate(lst)} bit = Bit(n) ans = 0 for t, x, y in query: x = dic[x] if t == 0: bit.add(x, y) else: y = dic[y] ans += bit.range_sum(x, y) print(ans)