【Codeforces-741D】Arpa’s letter-marked tree and Mehrdad’s Dokhtar-kosh paths(dsu on tree)

本文深入探讨了DSU on Tree算法,一种用于处理树形结构数据的高级数据结构和算法技术。通过详细的代码解析,文章展示了如何使用该算法解决复杂的问题,包括节点间的距离计算和子树操作。适合对数据结构和算法感兴趣的读者。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

在这里插入图片描述
套dsu on tree的套路
代码:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define ll long long
#define maxx 500005
#define INF 0x7f7f7f7f
using namespace std;
int n;
int head[maxx],to[maxx],_next[maxx];
int edge;
inline void addEdge(int x,int y)
{
    to[++edge]=y,_next[edge]=head[x],head[x]=edge;
}
int deep[maxx];
int _size[maxx];
int son[maxx];
int D[maxx];
void dfs1(int u,int fa)
{
    _size[u]=1;
    for(int i=head[u];i;i=_next[i])
    {
        int v=to[i];
        deep[v]=deep[u]+1;
        D[v]^=D[u];
        dfs1(v,u);
        _size[u]+=_size[v];
        if(_size[son[u]]<_size[v])son[u]=v;
    }
}
int l[maxx],r[maxx],_index;
int a[maxx];
void dfs2(int u,int fa)
{
    l[u]=++_index;
    a[_index]=u;
    for(int i=head[u];i;i=_next[i])
    {
        int v=to[i];
        dfs2(v,u);
    }
    r[u]=_index;
}
int c[1<<22];
void del(int u)
{
    for(int i=l[u];i<=r[u];i++)c[D[a[i]]]=0;
}
int ans[maxx];
int dfs(int u)
{
    for(int i=head[u];i;i=_next[i])
    {
        int v=to[i];
        if(v==son[u])continue;
        dfs(v);
        del(v);
        ans[u]=max(ans[u],ans[v]);
    }
    if(son[u])
    {
        dfs(son[u]);
        ans[u]=max(ans[u],ans[son[u]]);
        for(int i=head[u];i;i=_next[i])
        {
            int v=to[i];
            if(v!=son[u])
            {
                for(int i=l[v];i<=r[v];i++)
                {

                    if(c[D[a[i]]])ans[u]=max(ans[u],deep[a[i]]+c[D[a[i]]]-2*deep[u]);
                    for(int j=0;j<22;j++)
                        if(c[(1<<j)^D[a[i]]])
                            ans[u]=max(ans[u],deep[a[i]]+c[(1<<j)^D[a[i]]]-2*deep[u]);

                }
                for(int i=l[v];i<=r[v];i++)c[D[a[i]]]=max(c[D[a[i]]],deep[a[i]]);
            }
        }
    }
    if(c[D[u]])ans[u]=max(ans[u],c[D[u]]-deep[u]);
    for(int i=0;i<22;i++)
        if(c[(1<<i)^D[u]])ans[u]=max(ans[u],c[(1<<i)^D[u]]-deep[u]);
    c[D[u]]=max(c[D[u]],deep[u]);
}
int main()
{
    cin>>n;
    char s[5];
    int x;
    for(int i=2;i<=n;i++)
    {
        scanf("%d%s",&x,s);
        addEdge(x,i);
        D[i]=1<<(s[0]-'a');
    }
    //deep[1]=1;
    dfs1(1,0);
    dfs2(1,0);
    dfs(1);
    for(int i=1;i<n;i++)printf("%d ",ans[i]);
    printf("%d\n",ans[n]);
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值