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