【CodeForces - 570D】：Tree Requests【dsu on tree】

题目:

CodeForces - 570D：Tree Requests

题意:

给定一棵树，每个节点有一个字母，给出 m 次询问，每次询问能否将以 v 为根的子树中深度为 h 的节点上的字母任意拼接成一个回文串

分析:

容易得出字母数量为奇数的不同字母的个数 <= 1 时，答案为 yes，否则为 no

（1）首先可以按深度对节点分成不同的块，再跑一遍 dfs 序，那么子树就对应一段连续的区间，询问时只需要统计这个深度的块中一段连续的区间的节点信息即可，字母只有26个，求一遍前缀和，每次二分区间的左右端点即可

代码:

#include <bits/stdc++.h>

#define f first
#define s second
#define pii pair<int,int>
#define sz(x) (int)x.size()
#define all(x) (x).begin(),(x).end()
using namespace std;
const int maxn = 5e5+55;
int p,n,m,v,h,Time,st[maxn],ed[maxn],ans[maxn];
char s[maxn];
struct node{
    int l,r,id;
};
vector<node> q[maxn];
vector<int> g[maxn],d[maxn],w[maxn];
int sum[maxn][27];
void dfs(int x,int dep){
    st[x] = ++Time;
    d[dep].push_back(x),w[dep].push_back(st[x]);
    for(int i = 0;i < (int)g[x].size(); ++i){
        dfs(g[x][i],dep+1);
    }
    ed[x] = Time;
}
void solve(){
    for(int i = 1;i <= n; ++i){
        if(sz(q[i]) == 0) continue;
        for(int j = 0;j < sz(d[i]); ++j){
            for(int k = 0;k < 26; ++k) sum[j+1][k] = sum[j][k];
            sum[j+1][s[d[i][j]-1]-'a']++;
        }
        for(int j = 0;j < sz(q[i]); ++j){
            int L = lower_bound(all(w[i]),q[i][j].l)-w[i].begin();
            int R = upper_bound(all(w[i]),q[i][j].r)-w[i].begin();
            int cnt = 0;
            for(int k = 0;k < 26; ++k){
                if((sum[R][k]-sum[L][k])&1) cnt++;
            }
            if(cnt > 1) ans[q[i][j].id] = 1;
        }
    }
}
int main(){
    scanf("%d %d",&n,&m);
    for(int i = 2;i <= n; ++i){
        scanf("%d",&p);
        g[p].push_back(i);
    }
    scanf("%s",s); dfs(1,1);
    for(int i = 1;i <= m; ++i){
        scanf("%d %d",&v,&h);
        q[h].push_back((node){st[v],ed[v],i});
    }
    solve();
    for(int i = 1;i <= m; ++i){
        if(ans[i]) puts("No");
        else puts("Yes");
    }
    return 0;
}

（2）上面是离线实现的，发现可以通过对每一块不同的字母状压+异或的前缀和，最后统计1的个数后就能在线做了，由此启发离线后 dsu on tree 来做，维护重儿子的子树下每个深度的状压值即可

代码：

#include <bits/stdc++.h>
 
#define dp first
#define id second
#define pii pair<int,int>
using namespace std;
const int maxn = 5e5+55;
int n,m,p,v,h;
vector<int> g[maxn];
vector<pii> q[maxn];
int sz[maxn],son[maxn],d[maxn],ans[maxn],res[maxn];
char s[maxn];
void dfs(int x,int depth){
    sz[x] = 1; d[x] = depth;
    for(int i = 0;i < (int)g[x].size(); ++i){
        dfs(g[x][i],depth+1);
        sz[x] += sz[g[x][i]];
        if(sz[g[x][i]] > sz[son[x]]) son[x] = g[x][i];
    }
}
int cal(int x){
    int cnt = 0;
    for(int i = 0;i < 26; ++i) if(x&(1<<i)) cnt++;
    return cnt <= 1;
}
void deal(int x,int p){
    res[d[x]] ^= (1<<(s[x]-'a'));
    for(int i = 0;i < (int)g[x].size(); ++i){
        if(g[x][i] != p) deal(g[x][i],p);
    }
}
void dsu(int x,int op){
    for(int i = 0;i < (int)g[x].size(); ++i){
        if(g[x][i] != son[x]) dsu(g[x][i],0);
    }
    if(son[x]) dsu(son[x],1);
    deal(x,son[x]);
    for(int i = 0;i < (int)q[x].size(); ++i){
        ans[q[x][i].id] = cal(res[q[x][i].dp]);
    }
    if(!op) deal(x,0);
}
int main(){
    scanf("%d %d",&n,&m);
    for(int i = 2;i <= n; ++i){
        scanf("%d",&p);
        g[p].push_back(i);
    }
    scanf("%s",s+1); dfs(1,1);
    for(int i = 1;i <= m; ++i){
        scanf("%d %d",&v,&h);
        q[v].push_back(pii(h,i));
    }
    dsu(1,1);
    for(int i = 1;i <= m; ++i){
        if(ans[i]) puts("Yes");
        else puts("No");
    }
    return 0;
}