Submission #1113972

#TimeUsernameProblemLanguageResultExecution timeMemory
1113972spycoderytSynchronization (JOI13_synchronization)C++17
100 / 100
570 ms37980 KiB
#include <bits/stdc++.h>
#define f first
#define s second
using namespace std;
const int N = 3e5+5;
int st[N],in[N],out[N],p[33][N],fen[N],dp[2][N],child[N],dep[N];
vector<int> A[N];
int cnt = 1;
int n,m,q,a,b,x;
void add(int pos,int val) {
    for(int i = pos;i<N;i+=i&-i) fen[i]+=val;
}
int qry(int pos) {
    if(!pos)return 0;
    int res = 0;
    for(int i = pos;i>0;i-=i&-i) res += fen[i];
    return res;
}
void dfs(int u,int par = -1) {
    in[u]=cnt++;
    for(auto nxt : A[u])if(nxt!=par) {
        dep[nxt] = dep[u]+1;
        dfs(nxt,u);
        p[0][nxt] = u;
    }
    out[u]=cnt;
}
void dolca() {
    for(int i = 1;i<32;i++) {
        for(int j = 1;j<=n;j++) p[i][j] = p[i-1][p[i-1][j]];
    }
}
int fp(int x) {
    for(int i = 31;i>=0;i--) {
        if(p[i][x] && qry(in[p[i][x]]) == qry(in[x]))x=p[i][x];
    }
    return x;
}
int main() {
    cin>>n>>m>>q;
    vector<pair<int,int> > tmp;
    for(int i = 0;i<n-1;i++) {
        cin>>a>>b;
        A[a].push_back(b);A[b].push_back(a);
        tmp.push_back({a,b});
    }
    dfs(1);
    p[0][1] = 1;
    dolca();
    for(int i = 0;i<n-1;i++) {
        if(dep[tmp[i].f] < dep[tmp[i].s]) swap(tmp[i].f,tmp[i].s);
        child[i+1] = tmp[i].f;
    }
    for(int i = 1;i<=n;i++) add(in[i],1),add(out[i],-1),dp[0][i]=1;
    // for(int i = 1;i<=n;i++)cerr<<qry(i)-qry(i-1)<< " ";
    // cerr<<"\n";
    for(int i = 0;i<m;i++) {
        cin>>x;
        x = child[x];
        st[x] = 1-st[x];
        if(st[x]) {
            //turn on
            int anc = fp(p[0][x]);
            // cerr << x << " " << anc << "\n";
            add(in[x],-1),add(out[x],1);
            dp[0][anc] += dp[0][x] - dp[1][x];
        } else {
            // turn off
            int anc = fp(x);
            // cerr << x << " " << anc << "\n";
            add(in[x],1),add(out[x],-1);
            dp[0][x] = dp[1][x] = dp[0][anc];
        }
    }
    for(int i = 0;i<q;i++) {
        cin>>x;
        // if(st[x]) {
        //     st[x] = 1-st[x];
        //     int anc = fp(p[0][x]);
        //     add(in[x],1),add(out[x]+1,-1);
        //     dp[0][x] = dp[1][x] = dp[0][anc];
        // }
        // cout << dp[0][x] << "\n";
        // cerr << x << " " << fp(x) << "\n";
        cout << dp[0][fp(x)] << "\n";
    }
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...