結果
問題 | No.789 範囲の合計 |
ユーザー | AC2K |
提出日時 | 2023-05-04 01:34:29 |
言語 | C++23 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 226 ms / 1,000 ms |
コード長 | 4,202 bytes |
コンパイル時間 | 1,362 ms |
コンパイル使用メモリ | 97,624 KB |
実行使用メモリ | 34,944 KB |
最終ジャッジ日時 | 2024-05-01 17:59:08 |
合計ジャッジ時間 | 4,763 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,812 KB |
testcase_01 | AC | 2 ms
6,944 KB |
testcase_02 | AC | 206 ms
28,800 KB |
testcase_03 | AC | 74 ms
6,940 KB |
testcase_04 | AC | 226 ms
32,000 KB |
testcase_05 | AC | 191 ms
27,648 KB |
testcase_06 | AC | 202 ms
28,928 KB |
testcase_07 | AC | 71 ms
6,944 KB |
testcase_08 | AC | 210 ms
34,944 KB |
testcase_09 | AC | 183 ms
32,000 KB |
testcase_10 | AC | 169 ms
20,224 KB |
testcase_11 | AC | 158 ms
28,288 KB |
testcase_12 | AC | 156 ms
28,288 KB |
testcase_13 | AC | 2 ms
6,944 KB |
testcase_14 | AC | 2 ms
6,940 KB |
ソースコード
#line 2 "main.cpp" #include <cassert> #include <memory> namespace kyopro { /// @brief 動的セグメント木 template <class S, S (*op)(S, S), S (*e)()> class dynamic_segtree { public: dynamic_segtree(size_t n = 0) : n(n), root(nullptr) {} private: struct Node { S val; std::unique_ptr<Node> l, r; Node(const S& v) : val(v), l(nullptr), r(nullptr) {} }; using uptr = std::unique_ptr<Node>; uptr root; size_t n; public: void update(size_t p, const S& x) { assert(0 <= p && p < n); internal_update(root, 0, n, p, x); } void add(size_t p, const S& x) { assert(0 <= p && p < n); internal_add(root, 0, n, p, x); } S operator[](size_t p) const { assert(0 <= p && p < n); return internal_access(root, 0, n, p); } S prod(size_t l, size_t r) { assert(0 <= l && l <= r && r <= n); if (l == r) { return e(); } return internal_prod(root, 0, n, l, r); } private: void internal_update(uptr& p, size_t l, size_t r, size_t idx, const S& new_val) { if (p == nullptr) { p = std::make_unique<Node>(e()); } if (r - l == 1) { p->val = new_val; return; } size_t mid = (l + r) >> 1; if (idx < mid) { internal_update(p->l, l, mid, idx, new_val); } else { internal_update(p->r, mid, r, idx, new_val); } p->val = e(); if (p->left != nullptr) { p->val = op(p->left->val, p->val); } if (p->right != nullptr) { p->val = op(p->val, p->right->val); } } void internal_add(uptr& p, size_t l, size_t r, size_t idx, const S& new_val) { if (p == nullptr) { p = std::make_unique<Node>(e()); } if (r - l == 1) { p->val = op(p->val, new_val); return; } size_t mid = (l + r) >> 1; if (idx < mid) { internal_add(p->l, l, mid, idx, new_val); } else { internal_add(p->r, mid, r, idx, new_val); } p->val = e(); if (p->l != nullptr) { p->val = op(p->l->val, p->val); } if (p->r != nullptr) { p->val = op(p->val, p->r->val); } } S internal_access(uptr& p, size_t l, size_t r, size_t idx) { if (p == nullptr) { return e(); } if (r - l == 1) { return p->val; } size_t mid = (l + r) >> 1; if (idx < mid) { return internal_access(p->l, l, mid, idx); } else { return internal_access(p->r, mid, r, idx); } } S internal_prod(uptr& p, size_t l, size_t r, size_t L, size_t R) { if (p == nullptr || r <= L || R <= l) { return e(); } if (L <= l && r <= R) { return p->val; } size_t mid = (l + r) >> 1; return op(internal_prod(p->l, l, mid, L, R), internal_prod(p->r, mid, r, L, R)); } }; }; // namespace kyopro /// @docs docs/data-structure/dynamic_segtree.md #include <iostream> inline int op(int x, int y) { return x + y; } inline int e() { return 0; } int main() { const size_t n = 1000000001; kyopro::dynamic_segtree<int, op, e> seg(n); int q; std::cin >> q; long long ans = 0; while (q--) { int type; std::cin >> type; if (type == 0) { size_t x; long long y; std::cin >> x >> y; seg.add(x, y); } else { size_t l, r; std::cin >> l >> r; ans += seg.prod(l, r + 1); // cout << "ANS:" << seg.prod(l, r + 1) << '\n'; } /* cout << "["; for (int i = 0; i < 20; i++) { cout << seg[i] << ' '; } cout << "]\n"; */ } std::cout << ans << '\n'; }