#829199 (C++17) No.789 範囲の合計

提出ソース

結果

問題	No.789 範囲の合計
ユーザー	shu8Cream
提出日時	2022-12-29 20:05:25
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	73 ms / 1,000 ms
コード長	5,078 bytes
コンパイル時間	2,390 ms
コンパイル使用メモリ	210,604 KB
実行使用メモリ	6,820 KB
最終ジャッジ日時	2024-11-24 20:15:23
合計ジャッジ時間	3,782 ms
ジャッジサーバーID （参考情報）	judge4 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,816 KB
testcase_01	AC	2 ms 6,816 KB
testcase_02	AC	73 ms 6,820 KB
testcase_03	AC	54 ms 6,820 KB
testcase_04	AC	70 ms 6,816 KB
testcase_05	AC	52 ms 6,816 KB
testcase_06	AC	56 ms 6,816 KB
testcase_07	AC	33 ms 6,820 KB
testcase_08	AC	49 ms 6,816 KB
testcase_09	AC	46 ms 6,820 KB
testcase_10	AC	66 ms 6,816 KB
testcase_11	AC	42 ms 6,820 KB
testcase_12	AC	44 ms 6,820 KB
testcase_13	AC	2 ms 6,816 KB
testcase_14	AC	2 ms 6,816 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

/**
*    author:  shu8Cream
*    created: 28.12.2022 19:50:03
**/

#include <bits/stdc++.h>
using namespace std;
#define overload3(a,b,c,d,...) d
#define rep1(i,n) for (int i=0; i<(n); i++)
#define rep2(i,a,n) for (int i=(a); i<(n); i++)
#define rep(...) overload3(__VA_ARGS__, rep2, rep1)(__VA_ARGS__)
#define rrep1(i,n) for (int i=(n-1); i>=0; i--)
#define rrep2(i,a,n) for (int i=(n-1); i>=(a); i--)
#define rrep(...) overload3(__VA_ARGS__, rrep2, rrep1)(__VA_ARGS__)
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define sz(x) int((x).size())
#define pcnt __builtin_popcountll
using ll = long long;
using P = pair<ll,ll>;
template<typename T> using vc = vector<T>;
template<typename T> using vv = vc<vc<T>>;
using vi = vc<ll>;
using vvi = vv<ll>;
const int INF = 1e9;
const ll LINF = 8e18;
template<typename T>istream& operator>>(istream&i,vc<T>&v){rep(j,sz(v))i>>v[j];return i;}
template<typename T>string join(const T&v,const string& d=""){stringstream s;rep(i,sz(v))(i?s<<d:s)<<v[i];return s.str();}
template<typename T>ostream& operator<<(ostream&o,const vc<T>&v){if(sz(v))o<<join(v," ");return o;}
template<typename T1,typename T2>istream& operator>>(istream&i,pair<T1,T2>&v){return i>>v.first>>v.second;}
template<typename T1,typename T2>ostream& operator<<(ostream&o,const pair<T1,T2>&v){return o<<v.first<<","<<v.second;}
template<class T> inline bool chmax(T& a, T b) {if(a<b) { a=b;return true; } return false;}
template<class T> inline bool chmin(T& a, T b) {if(a>b) { a=b;return true; } return false;}

template <class T> string to_string(T s);
template <class S, class T> string to_string(pair<S, T> p);
string to_string(char c) { return string(1, c); }
string to_string(string s) { return s; }
string to_string(const char s[]) { return string(s); }

template <class T>
string to_string(T v) {
    if (v.empty()) return "{}";
    string ret = "{";
    for (auto x : v) ret += to_string(x) + ",";
    ret.back() = '}';
    return ret;
}
template <class S, class T>
string to_string(pair<S, T> p) {
    return "{" + to_string(p.first) + ":" + to_string(p.second) + "}";
}

void debug_out() { cout << endl; }

template <typename Head, typename... Tail>
void debug_out(Head H, Tail... T) {
    cout << to_string(H) << " ";
    debug_out(T...);
}

#ifdef _DEBUG
#define debug(...) debug_out(__VA_ARGS__)
#else
#define debug(...)
#endif

struct DynamicSegTree {
public:
    DynamicSegTree(int n): n(n),root(nullptr) {}
    void set(int pos, ll x) { set(root,0,n,pos,x); }
    ll get(int pos) { return get(root,0,n,pos); }
    ll prod(int l, int r) { return prod(root,0,n,l,r); }

private:
    struct node {
        int pos;
        ll val, sum;
        node *left;
        node *right;
        node(int pos, ll val)
            : pos(pos), val(val), sum(val), left(nullptr), right(nullptr) {}
    };
    int n;
    node *root;

    // 0-indexed, [l,r)
    void set(node *&t, int l, int r, int pos, ll x){
        if(t==nullptr) t = new node(pos,x); // 単位元で初期化
        if(pos==t->pos){
            t->val = x;
            t->sum = t->val;
            t->sum += (t->left?t->left->sum:0);
            t->sum += (t->right?t->right->sum:0);
            return;
        }
        // if(r-l==1){
        //     t->val += x; // =: 一点更新、 +=: 一点加算
        //     return;
        // }
        int mid = (l + r)>>1;
        if(pos<mid){
            if(t->pos < pos) { swap(t->pos,pos); swap(t->val,x); }
            set(t->left,l,mid,pos,x);
        }else{
            if(pos < t->pos) { swap(t->pos,pos); swap(t->val,x); }
            set(t->right,mid,r,pos,x);
        }

        // 葉以外のノードを更新
        // ここの演算は区間クエリに依存
        t->sum = t->val;
        t->sum += (t->left?t->left->sum:0);
        t->sum += (t->right?t->right->sum:0);
    }

    ll get(node *&t, int l, int r, int pos){
        if(t==nullptr) return 0; // 単位元を返す
        if(pos==t->pos) return t->val;
        int mid = (l + r)>>1;
        if(pos<mid) return get(t->left,l,mid,pos);
        else return get(t->right,mid,r,pos);
    }

    ll prod(node *&t, int sl, int sr, int l, int r){
        if(t==nullptr) return 0; // 単位元を返す
        if(sr<=l || r<=sl) return 0; // 単位元を返す
        if(l<=sl && sr<=r) return t->sum;
        int mid = (sl + sr)>>1;
        ll res = prod(t->left,sl,mid,l,r);
        if(l<=t->pos && t->pos < r) res += t->val;
        return res + prod(t->right,mid,sr,l,r);
    }
};

void YUKI789(){
    DynamicSegTree st(1000000000);
    int q; cin >> q;
    int op,x,l,r; ll y;
    ll ans = 0;
    for(;q--;){
        cin >> op;
        if(op==0){
            cin >> x >> y;
            ll v = st.get(x);
            st.set(x,v+y);
            debug("set", v+y);
        }else{
            cin >> l >> r;
            ans += st.prod(l,r+1);
            debug("ans", ans);
        }
    }
    cout << ans << endl;
}

int main() {
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    cout << fixed << setprecision(15);
    YUKI789();
}

yukicoder

結果

テストケース

ソースコード