poj3107:Godfather (树形dp)

题意

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;
}

猜你喜欢

转载自blog.csdn.net/sunshiness_s/article/details/79783066