【poj1655】Balancing Act

http://poj.org/problem?id=1655 (题目链接)

题意

　　给出一棵树，求树上一点使得使得删除一点后该树的最大子树最小。

solution

　　树的重心裸题。

　　随意取一点作为根节点，dfs维护当前节点的最大子树大小以及它的父亲子树大小，更新答案。

代码

// poj1655
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<ctime>
#define LL long long
#define inf 2147483640
#define eps 1e-10
#define Pi acos(-1.0)
#define free(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);
using namespace std;

const int maxn=20010;
struct edge {int to,next;}e[maxn<<2];
int head[maxn],vis[maxn],son[maxn],cnt,size,ans,n;

void link(int u,int v) {
	e[++cnt].to=v;e[cnt].next=head[u];head[u]=cnt;
	e[++cnt].to=u;e[cnt].next=head[v];head[v]=cnt;
}
void dfs(int x) {
	vis[x]=1,son[x]=0;
	int tmp=0;
	for (int i=head[x];i;i=e[i].next) if (!vis[e[i].to]) {
			dfs(e[i].to);
			son[x]+=son[e[i].to]+1;
			tmp=max(tmp,son[e[i].to]+1);
		}
	tmp=max(tmp,n-son[x]-1);
	if (tmp<size || (tmp==size && x<ans)) size=tmp,ans=x;
}
int main() {
	int T;scanf("%d",&T);
	while (T--) {
		scanf("%d",&n);
		memset(vis,0,sizeof(vis));memset(head,0,sizeof(head));
		cnt=ans=0;size=inf;
		for (int u,v,i=1;i<n;i++) {
			scanf("%d%d",&u,&v);
			link(u,v);
		}
		dfs(1);
		printf("%d %d\n",ans,size);
	}
    return 0;
}