记一个找树的重心/dfs

//求树的重心
int rcd[maxn];//记录去掉这个节点最大块的节点数
int dfs(int u,int fa)
{
    int i,v,t1,tmp;
    for(int i=head[u];i;i=edge[i].nxt)
    {
        v=edge[i].to;
        if(v==fa)continue;
        t1=dfs(v,u);
        if(t1>rcd[u])rcd[u]=t1;
        tmp+=t1;
        if(qsum-tmp<rcd[v])rcd[v]=qsum-tmp;
    }
    return tmp+1;
}
//找出重心的做法
int cnt;
int id1[2],mm=0x3f3f3f3f;
for(int i=1;i<=n;i++)
{
    if(rcd[i]<=mm)
    {   cnt=0;
        mm=rcd[i];
        id1[cnt++]=i;
    }
    else if(rcd[i]==mm)id1[cnt++]=i;//无根树的性质有两个树的重心,第二个
}

这个算法我理解的是通过不停的回溯,t1为数的子节点的个数,如果大于就直接更新这个值,而父亲就是通过qsum-tmp来实现,
//返回的tmp+1则是当前节点的个数,嗯ok,找的就是对的rcd数组

猜你喜欢

转载自blog.csdn.net/qq_42193011/article/details/81608782