poj 1655 Balancing Act【树的重心】

题目原文
题意：

代码：

/* 使用前初始化G
* 时间复杂度O(n)
*/
vector<int> G[20005];
int n;
int dp[20005],w[20005];
int dfs(int u,int f)
{
if(dp[u]!=-1) return dp[u];
int res=0,sum=0;
for(int i=0;i<G[u].size();i++)
{
int to=G[u][i];
if(to==f) continue;
res=max(res,dfs(to,u));
sum+=dfs(to,u);
}
res=max(n-sum-1,res);
w[u]=res;
return dp[u]=sum+1;
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
memset(dp,-1,sizeof dp);
for(int i=0;i<20005;i++) G[i].clear();
scanf("%d",&n);
for(int i=1;i<n;i++)
{
int u,v;
scanf("%d%d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
dfs(1,-1);
int Min=INF,Mini;
for(int i=1;i<=n;i++)
{
if(w[i]<Min) Min=w[i],Mini=i;
}
printf("%d %d\n",Mini,Min);
}
return 0;
}