POJ.3107 Godfather (树形DP 树的重心)

POJ.3107 Godfather (树形DP 树的重心)

标签（空格分隔）： 动态规划

---

题意分析

给出一棵树，求出树的重心，树的重心可能有多个。
很裸，记一下最小联通块节点个数$min$，然后对dp扫一遍输出就可以。
注意数据范围是5w。

代码总览

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#define nmax 55555
using namespace std;
struct edge{
    int to,next;
}edg[nmax*2];
int head[nmax],sz[nmax];
int dp[nmax];
int tot = 0,ans = 0;
int t,n;
void add(int u, int v){
    edg[tot].to = v;
    edg[tot].next = head[u];
    head[u] = tot++;
}
void dfs(int rt, int fa){
    sz[rt] = 1;
    dp[rt] = 0;
    int tmp = 0;
    for(int i = head[rt]; i!=-1 ;i = edg[i].next){
        int nxt = edg[i].to;
        if(nxt != fa){
            dfs(nxt,rt);
            sz[rt]+=sz[nxt];
            tmp = max(tmp,sz[nxt]);
        }
    }
    tmp = max(tmp,n-sz[rt]);
    dp[rt] = tmp;
    if(tmp < ans) ans = tmp;
}
void init(){
    memset(head,-1,sizeof head);
    tot = 0;
    ans = 1e9+7;
}
int main()
{
    //freopen("im.txt","r",stdin);
    while(scanf("%d",&n) != EOF){
        init();
        for(int i = 0;i<n-1;++i){
            int x,y;
            scanf("%d %d",&x,&y);
            add(x,y);
            add(y,x);
        }
        dfs(1,0);
        bool isfirst = true;
        //printf("%d\n",ans);
        for(int i = 1;i<=n;++i){
            if(dp[i] == ans ){
                if(isfirst){printf("%d",i);isfirst = false;}
                else printf(" %d",i);
            }
        }
        printf("\n");
    }
    return 0;
}