結果

問題 No.789 範囲の合計
ユーザー AC2KAC2K
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

#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';
}
0