poj 2378 树型DP

题意：

n个点组成了一棵树，去掉某一个点会出现一些个连通块，问去掉哪些个点会使剩下的连通块中  最大的那个连通块小于等于n。

题解：

和poj 3107 一样。。。。

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int sum[10010],max1[10010],list[10010],n,pos;
struct P{
	int u,v,next;
}point[20010];
struct It{
	int numb,ki;
}it[10010];
void add(int son,int fa){
	point[pos].u=fa;
	point[pos].v=son;
	point[pos].next=list[fa];
	list[fa]=pos++;
}
void dfs(int son,int fa){
	int now=list[son];
	sum[son]=1;
	while(now!=-1){
		if(point[now].v==fa){
			now=point[now].next;
			continue;
		}
		dfs(point[now].v,son);
		sum[son]+=sum[point[now].v];
		max1[son]=max(max1[son],sum[point[now].v]);
		now=point[now].next;
	}
	return;
}
void dfs1(int son,int fa){
	if(fa==0){
		it[son].ki=max1[son];
	}
	else{
		it[son].ki=max(max1[son],n-sum[son]);
	}
	int now=list[son];
	while(now!=-1){
		if(point[now].v==fa){
			now=point[now].next;
			continue;
		}
		dfs1(point[now].v,son);
		now=point[now].next;
	}
	return;
}
int cmp(It a,It b){
	if(a.ki<b.ki){
		return a.ki<b.ki;
	}
	else if(a.ki==b.ki){
		return a.numb<b.numb;
	}
	else
		return a.ki<b.ki;
}
int main(){
	while(scanf("%d",&n)!=EOF){
		pos=0;memset(list,-1,sizeof(list));
		for(int i=1;i<n;i++){
			int a,b;
			scanf("%d%d",&a,&b);
			add(a,b);
			add(b,a);
			it[i].numb=i;
		}
		it[n].numb=n;
		memset(sum,0,sizeof(sum));
		memset(max1,0,sizeof(max1));
		dfs(1,0);
		dfs1(1,0);
//		for(int i=1;i<=n;i++){
//			printf("! sum=%d max1=%d ki=%d\n",sum[i],max1[i],it[i].ki);
//		}
		sort(it+1,it+n+1,cmp);
		int k=it[1].ki;
		for(int i=1;i<=n&&it[i].ki==k&&it[i].ki<=n/2;i++){
			printf("%d\n",it[i].numb);
		}
	}
	return 0;
}