DSU on tree CodeForces - 741D

dsu on tree 

主要还是去学了一下这个东西，大概就是利用的轻重链的性质，把暴力的复杂度优化到了（nlogn）。非常神奇的算法。

题目还是略板的。

题目意思就是问每棵子树上有多少权值异或和为0或者二进制最多1个1的链。

#include <bits/stdc++.h>

using namespace std;
#define N 500005
#define go(i,a,b) for(int i=(a);i<=(b);i++)
#define inf 0x3f3f3f3f
struct no{
    int to,n,w;
};no eg[N*2];
int h[N],sz[N],son[N],fi[N],ed[N],id[N],dep[N],v[N],ans[N],f[1<<22],n,to, cnt, tot;
void add(int u,int to,int w){
    eg[++tot]={to,h[u],w};h[u]=tot;
    eg[++tot]={u,h[to],w};h[to]=tot;
}
void dfs(int u,int fa){

    id[fi[u]=++cnt]=u;sz[u]=1;
    for(int i=h[u];i;i=eg[i].n){
        int to=eg[i].to;
        if(to==fa)continue;
        v[to]=v[u]^eg[i].w;
        dep[to]=dep[u]+1;
        dfs(to,u);
        sz[u]+=sz[to];
        if(sz[to]>sz[son[u]])son[u]=to;
    }
    ed[u]=cnt;
}
void dsudfs(int u,int fa,int op=0){
    //cout<<u<<' '<<fa<<endl;
    for(int i=h[u];i;i=eg[i].n){
        int to=eg[i].to;
        if(to==fa||to==son[u])continue;
        dsudfs(to,u);ans[u]=max(ans[u],ans[to]);
    }
    if(son[u])dsudfs(son[u],u,1),ans[u]=max(ans[u],ans[son[u]]);
    go(i,0,21)ans[u]=max(ans[u],f[v[u]^(1<<i)]-dep[u]);
    ans[u]=max(ans[u],f[v[u]]-dep[u]);f[v[u]]=max(f[v[u]],dep[u]);

    for(int i=h[u];i;i=eg[i].n){
        int to=eg[i].to;
        if(to==fa||to==son[u])continue;
        go(j,fi[to],ed[to]){
            go(ii,0,21)ans[u]=max(ans[u],f[v[id[j]]^(1<<ii)]+dep[id[j]]-dep[u]*2);
            ans[u]=max(ans[u],f[v[id[j]]]+dep[id[j]]-dep[u]*2);
        }
        go(j,fi[to],ed[to])f[v[id[j]]]=max(f[v[id[j]]],dep[id[j]]);
    }
    if(!op)go(i,fi[u],ed[u])f[v[id[i]]]=-inf;
}
char s[20];
int main()
{
    cin>>n;
    memset(f,-inf,sizeof f);//cout<<f[0]<<endl;
    go(i,2,n)scanf("%d %s",&to,s),add(i,to,1<<(s[0]-'a'));
    dfs(1,0);dsudfs(1,0);
    go(i,1,n)printf("%d%c",ans[i], i==n?'\n':' ');
    return 0;
}