結果

問題 No.789 範囲の合計
ユーザー kuhakukuhaku
提出日時 2021-09-27 11:42:50
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 930 ms / 1,000 ms
コード長 4,075 bytes
コンパイル時間 2,151 ms
コンパイル使用メモリ 207,056 KB
実行使用メモリ 48,368 KB
最終ジャッジ日時 2023-09-20 11:35:06
合計ジャッジ時間 10,561 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,380 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 806 ms
44,312 KB
testcase_03 AC 154 ms
4,380 KB
testcase_04 AC 930 ms
45,164 KB
testcase_05 AC 743 ms
44,388 KB
testcase_06 AC 789 ms
44,296 KB
testcase_07 AC 344 ms
4,376 KB
testcase_08 AC 877 ms
48,368 KB
testcase_09 AC 799 ms
45,136 KB
testcase_10 AC 586 ms
26,840 KB
testcase_11 AC 686 ms
44,528 KB
testcase_12 AC 667 ms
44,480 KB
testcase_13 AC 2 ms
4,376 KB
testcase_14 AC 2 ms
4,384 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#line 1 "a.cpp"
#define PROBLEM ""
#line 2 "/home/kuhaku/kuhaku/github/atcoder-lib/lib/template/template.hpp"
#include <bits/stdc++.h>
using namespace std;
template <class T, class U>
bool chmax(T &a, const U &b) {
    return a < b ? a = b, true : false;
}
template <class T, class U>
bool chmin(T &a, const U &b) {
    return b < a ? a = b, true : false;
}
constexpr int64_t INF = 1000000000000000003;
constexpr int Inf = 1000000003;
constexpr int MOD = 1000000007;
constexpr int MOD_N = 998244353;
constexpr double EPS = 1e-7;
const double PI = acos(-1.0);
#line 3 "/home/kuhaku/kuhaku/github/atcoder-lib/lib/template/atcoder.hpp"
using ll = int64_t;
using ld = long double;
#define FOR(i, m, n) for(int i = (m); i < (n); ++i)
#define FORR(i, m, n) for(int i = (m)-1; i >= (n); --i)
#define rep(i, n) FOR(i, 0, n)
#define repn(i, n) FOR(i, 1, n+1)
#define repr(i, n) FORR(i, n, 0)
#define repnr(i, n) FORR(i, n+1, 1)
#define all(s) (s).begin(), (s).end()
template <class T>
istream &operator>>(istream &is, vector<T> &v) { for (T &i : v) is>>i; return is; }
template <class T>
ostream &operator<<(ostream &os, const vector<T> &v) {
    for (auto it=v.begin(); it!=v.end(); ++it) { os<<(it==v.begin()?"":" ")<<*it; } return os;
}
template <class Head, class... Tail>
void co(Head&& head, Tail&&... tail) {
    if constexpr(sizeof...(tail)==0) cout<<head<<'\n'; else cout<<head<<' ',co(forward<Tail>(tail)...);
}
template <class Head, class... Tail>
void ce(Head&& head, Tail&&... tail) {
    if constexpr(sizeof...(tail)==0) cerr<<head<<'\n'; else cerr<<head<<' ',ce(forward<Tail>(tail)...);
}
template<typename T, typename... Args>
auto make_vector(T x, int arg, Args ...args) {
    if constexpr(sizeof...(args)==0) return vector<T>(arg, x); else return vector(arg,make_vector<T>(x, args...));
}
void sonic() { ios::sync_with_stdio(false); cin.tie(nullptr); }
void setp(const int n) { cout << fixed << setprecision(n); }
#line 3 "a.cpp"

template <class T, class F>
struct segment_tree {
    int N;
    const T e;
    const F op;
    unordered_map<int, T> data;

    segment_tree() {}
    segment_tree(int _n, T _e, F &&_op) : e(_e), op(_op) {
        init(_n);
    }
    segment_tree(int _n, T _e, const F &_op) : e(_e), op(_op) {
        init(_n);
    }

    const T &operator[](int i) const {
        return data[i + N];
    }
    T at(int k) const {
        assert(0 <= k && k < N);
        return data[k + N];
    }
    T get(int k) const {
        return at(k);
    }

    void init(int n) {
        for (N = 1; N < n; N <<= 1) {}
    }

    template <class U>
    void build(const vector<U> &v) {
        int n = v.size();
        for (int i = 0; i < n; ++i) data[N + i] = T(v[i]);
        for (int i = N - 1; i >= 1; --i)
            data[i] = op(data[i * 2], data[i * 2 + 1]);
    }

    void update(int k, T x) {
        data[k += N] = x;
        while ((k >>= 1) >= 1) {
            if (data.count(k * 2) && data.count(k * 2 + 1))
                data[k] = op(data[k * 2], data[k * 2 + 1]);
            else if (data.count(k * 2))
                data[k] = data[k * 2];
            else if (data.count(k * 2 + 1))
                data[k] = data[k * 2 + 1];
        }
    }

    T query(int a, int b) {
        T l = e, r = e;
        for (a += N, b += N; a < b; a >>= 1, b >>= 1) {
            if (a & 1) {
                if (data.count(a))
                    l = op(l, data[a]);
                ++a;
            }
            if (b & 1) {
                --b;
                if (data.count(b))
                    r = op(data[b], r);
            }
        }
        return op(l, r);
    }
};

int main(void) {
    sonic();
    setp(10);
    auto op = [](auto a, auto b) {
        return a + b;
    };
    segment_tree st(1e9 + 1, 0L, op);

    int n;
    cin >> n;
    ll ans = 0;
    unordered_map<int, ll> mp;
    rep(i, n) {
        int a, b, c;
        cin >> a >> b >> c;
        if (!a) {
            st.update(b, mp[b] + c);
            mp[b] += c;
        } else {
            ans += st.query(b, c + 1);
        }
    }
    co(ans);

    return 0;
}
0