poj3107：Godfather （树形dp）

最新推荐文章于 2021-02-14 14:20:00 发布

overcastt

最新推荐文章于 2021-02-14 14:20:00 发布

阅读量191

点赞数

分类专栏： dp

本文链接：https://blog.csdn.net/sunshiness_s/article/details/79783066

版权

dp 专栏收录该内容

30 篇文章

订阅专栏

本文介绍了一种通过求解树的重心来找到删除哪个节点可以使形成的最大子树节点数最小的方法，并提供了一段C++代码实现。

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

题意

poj3107
树上删去一个点以后，使形成最大的子树的结点数最小，求所有这样的点

题解

这题啊..是求树的重心【捂脸】
树的重心是啥呢~就是删除这个点后最大连通块的结点数最小
咋求呢
删除每个点后的最大连通块无非就是它的每个儿子的大小，或者是总结点数减去它所在的子树的大小…

代码

#include <cstdio>
#include <algorithm>
using namespace std;
#define N 51000
#define inf 0x7fffffff
int h[N],d[N],size[N],ans[N],mn=inf,n,num=0,tot;
struct node1{int x,y,next;}mp[N<<1];
void insert(int x,int y){
    mp[++num].x=x;mp[num].y=y;mp[num].next=h[x];h[x]=num;
    mp[++num].x=y;mp[num].y=x;mp[num].next=h[y];h[y]=num;
}
void dfs(int u){
    size[u]=1;int son=0;
    for(int i=h[u];i;i=mp[i].next){
        int v=mp[i].y;if(d[v]) continue;
        d[v]=d[u]+1;dfs(v);
        size[u]+=size[v];son=max(son,size[v]);
    }son=max(son,n-size[u]);
    if(son<mn){mn=son;tot=0;ans[++tot]=u;}
    else if(son==mn) ans[++tot]=u;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        int x,y;scanf("%d%d",&x,&y);
        insert(x,y);
    }d[1]=1;dfs(1);
    sort(ans+1,ans+1+tot);
    for(int i=1;i<=tot;i++) printf("%d ",ans[i]);
    return 0;
}