Arpa’s letter-marked tree and Mehrdad’s Dokhtar-kosh paths CodeForces - 741D

http://codeforces.com/contest/741/problem/D

问每棵子树上最长的满足条件的简单路径的长度

22个字母状压一下 保存每个节点到祖先的异或值 这样两节点u和v异或时 从lca到祖先这一部分的值会被抵消 巧妙的思路

然后就可以用树上启发式合并搞 保存每个异或值对应的最大深度 看一棵子树上异或值满足条件且经过子树根节点的两点的最远距离 类似树分治的思想

一开始写的时候 对于一棵子树u直接把所有子树v1 v2...vk的点都先更新 再全部查询 但是这样同一棵子树vi上的点也会被算上 应该先查询玩一棵子树 再更新该子树

 

还有几个比较恶心的坑点 debug整整一天 详见注释

 

#include <bits/stdc++.h>
using namespace std;
const int maxn=5e5+10;
const int maxm=5e6+10;

struct node
{
    int v,next;
};

node edge[maxn];
int book[maxm];
int first[maxn],val[maxn],sum[maxn],son[maxn],deep[maxn],mp1[maxn],mp2[maxn],ans[maxn];
int pre[50];
int n,num;

void addedge(int u,int v)
{
    edge[num].v=v;
    edge[num].next=first[u];
    first[u]=num++;
}

void init()
{
    int i;
    pre[0]=1;
    for(i=1;i<=30;i++) pre[i]=2*pre[i-1];
}

void dfsI(int cur,int fa)
{
    int i,v;
    val[cur]=(val[cur]^val[fa]),sum[cur]=1,son[cur]=-1;
    num++;
    mp1[cur]=num,mp2[num]=cur;
    for(i=first[cur];i!=-1;i=edge[i].next){
        v=edge[i].v;
        if(v!=fa){
            deep[v]=deep[cur]+1;
            dfsI(v,cur);
            sum[cur]+=sum[v];
            if(son[cur]==-1||sum[son[cur]]<sum[v]) son[cur]=v;
        }
    }
}

void dfsII(int cur,int fa,int flag)
{
    int i,j,k,v;
    for(i=first[cur];i!=-1;i=edge[i].next){
        v=edge[i].v;
        if(v!=fa&&v!=son[cur]) dfsII(v,cur,0);
    }
    if(son[cur]!=-1) dfsII(son[cur],cur,1);
    if(book[val[cur]]) ans[cur]=max(ans[cur],book[val[cur]]-deep[cur]);
    for(k=0;k<22;k++){
        if(book[(val[cur])^(pre[k])]) ans[cur]=max(ans[cur],book[(val[cur])^(pre[k])]-deep[cur]);//book[-]为零要特判
    }
    book[val[cur]]=max(book[val[cur]],deep[cur]);
    for(i=first[cur];i!=-1;i=edge[i].next){
        v=edge[i].v;
        if(v!=fa&&v!=son[cur]){
            for(j=mp1[v];j<=mp1[v]+sum[v]-1;j++){
                if(book[val[mp2[j]]]) ans[cur]=max(ans[cur],deep[mp2[j]]+book[val[mp2[j]]]-2*deep[cur]);
                for(k=0;k<22;k++){
                    if(book[(val[mp2[j]])^(pre[k])]) ans[cur]=max(ans[cur],deep[mp2[j]]+book[(val[mp2[j]])^(pre[k])]-2*deep[cur]);
                }
            }
            for(j=mp1[v];j<=mp1[v]+sum[v]-1;j++){
                book[val[mp2[j]]]=max(book[val[mp2[j]]],deep[mp2[j]]);
            }
        }
    }
    if(!flag){
        for(j=mp1[cur];j<=mp1[cur]+sum[cur]-1;j++){
            book[val[mp2[j]]]=0;
        }
    }
}

void dfsIII(int cur,int fa)
{
    int i,v;
    for(i=first[cur];i!=-1;i=edge[i].next){
        v=edge[i].v;
        if(v!=fa){
            //ans[cur]=max(ans[cur],ans[v]); 此时的ans[cur]中是没有重孩子节点的答案的
            dfsIII(v,cur);
            ans[cur]=max(ans[cur],ans[v]);//递归结束后才能带回重孩子节点的答案 ans[cur]必须在遍历完子树后再统计
        }
    }
}

int main()
{
    int i,u,v;
    char ch[10];
    init();
    scanf("%d",&n);
    memset(first,-1,sizeof(first));
    for(v=2;v<=n;v++){
        scanf("%d%s",&u,ch);
        addedge(u,v);
        val[v]=pre[ch[0]-'a'];
    }
    num=0;
    dfsI(1,0);
    dfsII(1,0,0);
    dfsIII(1,0);
    for(i=1;i<=n;i++) printf("%d ",ans[i]);
    printf("\n");
    return 0;
}