結果
問題 | No.789 範囲の合計 |
ユーザー | toyuzuko |
提出日時 | 2020-05-17 13:35:00 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
TLE
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 1,132 bytes |
コンパイル時間 | 406 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 50,336 KB |
最終ジャッジ日時 | 2024-09-25 05:17:51 |
合計ジャッジ時間 | 10,891 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 34 ms
10,880 KB |
testcase_01 | AC | 30 ms
10,880 KB |
testcase_02 | AC | 935 ms
43,136 KB |
testcase_03 | AC | 642 ms
20,864 KB |
testcase_04 | AC | 922 ms
42,912 KB |
testcase_05 | AC | 877 ms
42,812 KB |
testcase_06 | AC | 891 ms
42,988 KB |
testcase_07 | AC | 644 ms
21,120 KB |
testcase_08 | AC | 740 ms
34,108 KB |
testcase_09 | AC | 729 ms
31,904 KB |
testcase_10 | TLE | - |
testcase_11 | AC | 852 ms
42,652 KB |
testcase_12 | AC | 826 ms
42,516 KB |
testcase_13 | AC | 29 ms
10,752 KB |
testcase_14 | AC | 29 ms
10,880 KB |
ソースコード
class BinaryIndexedTree(): #1-indexed def __init__(self, n): self.n = n self.tree = [0 for _ in range(n + 1)] def sum(self, idx): res = 0 while idx: res += self.tree[idx] idx -= idx & -idx return res def add(self, idx, x): while idx <= self.n: self.tree[idx] += x idx += idx & -idx def bisect_left(self, x): if x <= 0: return 0 res, tmp = 0, 2**((self.n).bit_length() - 1) while tmp: if res + tmp <= self.n and self.tree[res + tmp] < x: x -= self.tree[res + tmp] res += tmp tmp >>= 1 return res + 1 N = int(input()) Q = [] V = set() for i in range(N): q, x, y = map(int, input().split()) if q == 0: V.add(x) else: V.add(x) V.add(y) Q.append((q, x, y)) comp = {v : k for k, v in enumerate(sorted(V), 2)} res = 0 bit = BinaryIndexedTree(2 * N + 1) for q, x, y in Q: if q == 0: bit.add(comp[x], y) else: res += bit.sum(comp[y]) - bit.sum(comp[x] - 1) print(res)