結果

問題 No.1790 Subtree Deletion
ユーザー shiroha_F14shiroha_F14
提出日時 2021-12-11 14:56:18
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 375 ms / 3,000 ms
コード長 1,943 bytes
コンパイル時間 1,964 ms
コンパイル使用メモリ 181,756 KB
実行使用メモリ 20,480 KB
最終ジャッジ日時 2024-09-15 14:06:30
合計ジャッジ時間 6,790 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 5 ms
6,784 KB
testcase_01 AC 4 ms
6,784 KB
testcase_02 AC 4 ms
6,784 KB
testcase_03 AC 351 ms
20,352 KB
testcase_04 AC 349 ms
20,352 KB
testcase_05 AC 350 ms
20,480 KB
testcase_06 AC 345 ms
20,480 KB
testcase_07 AC 341 ms
20,352 KB
testcase_08 AC 106 ms
6,784 KB
testcase_09 AC 375 ms
20,352 KB
testcase_10 AC 332 ms
20,352 KB
testcase_11 AC 332 ms
20,352 KB
testcase_12 AC 206 ms
19,968 KB
testcase_13 AC 235 ms
18,816 KB
testcase_14 AC 99 ms
9,984 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/lazysegtree>
using namespace atcoder;

using ll = long long;


// Euler Tour
vector<int> vis;
vector<ll> edges(1, 0);
vector<vector<pair<int, ll>>> g(100000);
vector<int> in(100000);
vector<int> out(100000);
vector<int> seen(100000, false);
int t = 0;

void dfs(int p){
    in[p] = t;
    t++;
    vis.push_back(p);
    seen[p] = true;
    
    for(auto v : g[p]){
        if(seen[v.first]) continue;
        edges.push_back(v.second);
        dfs(v.first);
        vis.push_back(p);
        edges.push_back(0);
    }

    t++;
    out[p] = t;
}

// lazy segtree
ll ID = 8000000000000000000;
ll op(ll a, ll b){
    return a ^ b;
}
ll e(){
    return 0;
}
ll mapping(ll f, ll x){
    if(f == ID){
        return x;
    }else{
        return f;
    }
}
ll composition(ll f, ll g){
    if(f == ID){
        return g;
    }else{
        return f;
    }
}
ll id(){
    return ID;
}

// main
int main(){
    int n; cin >> n;
    assert(2 <= n && n <= 100000);
    for(int i = 0; i < n - 1; i++){
        int l, r; cin >> l >> r;
        assert(1 <= l && l <= n && 1 <= r && r <= n);
        ll a; cin >> a;
        assert(0 <= a && a <= 1000000000000000000);
        g[l - 1].push_back({r - 1, a});
        g[r - 1].push_back({l - 1, a});
    }

    vis.reserve(n * 2);
    edges.reserve(n * 2);

    dfs(0);
    edges.push_back(0);

    int m = vis.size();
    lazy_segtree<ll, op, e, ll, mapping, composition, id> seg(edges);

    int q; cin >> q;
    assert(1 <= q && q <= 100000);
    for(int i = 0; i < q; i++){
        int t; cin >> t;
        assert(t == 1 || t == 2);
        int x; cin >> x;
        if(t == 1){
            // delete
            assert(2 <= x && x <= n);
            seg.apply(in[x - 1], out[x - 1], 0);
        }else{
            // f(x)
            assert(1 <= x && x <= n);
            cout << seg.prod(in[x - 1] + 1, out[x - 1]) << endl;
        }
    }
}
0