結果

問題 No.900 aδδitivee
ユーザー face4face4
提出日時 2020-01-13 17:10:16
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 328 ms / 2,000 ms
コード長 2,570 bytes
コンパイル時間 1,227 ms
コンパイル使用メモリ 83,552 KB
実行使用メモリ 31,496 KB
最終ジャッジ日時 2024-05-10 03:39:38
合計ジャッジ時間 11,692 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
6,812 KB
testcase_01 AC 4 ms
6,940 KB
testcase_02 AC 4 ms
6,940 KB
testcase_03 AC 4 ms
6,944 KB
testcase_04 AC 4 ms
6,944 KB
testcase_05 AC 4 ms
6,944 KB
testcase_06 AC 4 ms
6,940 KB
testcase_07 AC 310 ms
23,804 KB
testcase_08 AC 307 ms
23,884 KB
testcase_09 AC 307 ms
23,772 KB
testcase_10 AC 310 ms
23,828 KB
testcase_11 AC 305 ms
23,728 KB
testcase_12 AC 306 ms
23,848 KB
testcase_13 AC 305 ms
23,856 KB
testcase_14 AC 306 ms
23,764 KB
testcase_15 AC 305 ms
23,744 KB
testcase_16 AC 316 ms
23,892 KB
testcase_17 AC 309 ms
23,772 KB
testcase_18 AC 306 ms
23,720 KB
testcase_19 AC 306 ms
23,792 KB
testcase_20 AC 308 ms
23,808 KB
testcase_21 AC 303 ms
23,812 KB
testcase_22 AC 294 ms
31,356 KB
testcase_23 AC 294 ms
31,356 KB
testcase_24 AC 297 ms
31,496 KB
testcase_25 AC 296 ms
31,364 KB
testcase_26 AC 295 ms
31,392 KB
testcase_27 AC 328 ms
31,412 KB
testcase_28 AC 323 ms
31,492 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<iostream>
#include<vector>
using namespace std;

typedef long long ll;

class STlazy{
private:
    int n;
    vector<ll> node, lazy;
    vector<int> plus, minus;

public:
    STlazy(vector<int> v, vector<int> p){
        int siz = v.size();
        n = 1;
        while(n < siz)  n *= 2;
        node.resize(2*n-1, 0);
        lazy.resize(2*n-1, 0);
        plus.resize(2*n-1, 0);
        minus.resize(2*n-1, 0);
        for(int i = 0; i < siz; i++)    node[n-1+i] = v[i], plus[n-1+i] = p[i], minus[n-1+i] = 1-p[i];
        for(int i = n-2; i >= 0; i--)   node[i] = node[2*i+1] + node[2*i+2], plus[i] = plus[2*i+1]+plus[2*i+2], minus[i] = minus[2*i+1]+minus[2*i+2];
    }

    void eval(int k, int l, int r){
        if(lazy[k] == 0)    return;
        node[k] += lazy[k]*(plus[k]-minus[k]);
        if(r-l > 1){
            lazy[2*k+1] += lazy[k];
            lazy[2*k+2] += lazy[k];
        }
        lazy[k] = 0;
    }

    void add(int a, int b, ll x, int k=0, int l=0, int r=-1){
        if(r < 0)   r = n;
        eval(k, l, r);
        if(r <= a || b <= l)    return;
        if(a <= l && r <= b){
            lazy[k] += x;
            eval(k, l, r);
            return;
        }
        add(a, b, x, 2*k+1, l, (l+r)/2);
        add(a, b, x, 2*k+2, (l+r)/2, r);
        node[k] = node[2*k+1] + node[2*k+2];
    }

    ll query(int a, int b, int k=0, int l=0, int r=-1){
        if(r < 0)   r = n;
        if(r <= a || b <= l)    return 0;
        eval(k, l, r);
        if(a <= l && r <= b)    return node[k];
        ll lx = query(a, b, 2*k+1, l, (l+r)/2);
        ll rx = query(a, b, 2*k+2, (l+r)/2, r);
        return lx+rx;
    }
};

typedef pair<int,int> pii;

int k = 0;
vector<int> etour, pl;
vector<pii> v[100000];
int be[100000], en[100000];

void euler_tour(int x, int p){
    be[x] = k;
    for(pii P : v[x]){
        if(P.first == p)   continue;
        pl.push_back(1);
        etour.push_back(P.second);
        k++;
        euler_tour(P.first, x);
        pl.push_back(0);
        etour.push_back(-P.second);
        k++;
    }
    en[x] = k;
}

int main(){
    int n;
    cin >> n;
    for(int i = 0; i < n-1; i++){
        int a, b, c;
        cin >> a >> b >> c;
        v[a].push_back({b,c});
    }
    euler_tour(0, -1);
    STlazy seg(etour, pl);
    int q;
    cin >> q;
    while(q--){
        int op, a, b;
        cin >> op;
        if(op == 1){
            cin >> a >> b;
            seg.add(be[a], en[a], b);
        }else{
            cin >> a;
            cout << seg.query(0, be[a]) << endl;
        }
    }
    return 0;
}
0