結果

問題 No.900 aδδitivee
ユーザー face4face4
提出日時 2020-01-13 16:45:14
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,741 bytes
コンパイル時間 1,088 ms
コンパイル使用メモリ 83,532 KB
実行使用メモリ 31,536 KB
最終ジャッジ日時 2024-05-10 02:59:47
合計ジャッジ時間 9,621 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
5,888 KB
testcase_01 AC 4 ms
5,888 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
権限があれば一括ダウンロードができます

ソースコード

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); // 最初の呼び出しが必ずeval(0,0,n)となるので漏れなく上から値の伝播がなされている
        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);
        euler_tour(P.first, x);
        pl.push_back(0);
        etour.push_back(-P.second);
    }
    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