POJ--1655--Balancing Act--简单树形DP

注意建树过程

#include<iostream>
#include<cstring>
#include<cstdio>
#include<vector>
#define maxn 20001
using namespace std;
vector<int> g[maxn];
int p[maxn];
int banlance[maxn];
int n;
void init()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		p[i]=-1;
		g[i].clear();
	}
	int x,y;
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		g[x].push_back(y);
		g[y].push_back(x);
	}
}
void build(int u)
{
	for(int i=0;i<g[u].size();i++)
	{
		int v=g[u][i];
		if(p[v]!=-1)continue;
		p[v]=u;
		build(v);
	}
}

int dfs(int u)
{
	int sum=0;
	int res=0;
	int _max=0;
	for(int i=0;i<g[u].size();i++)
	{
		int v=g[u][i];
		if(v==p[u])continue;
		res=dfs(v);
		sum+=res;
		_max=max(_max,res);
	}
	_max=max(_max,n-1-sum);
	banlance[u]=_max;
	return sum+1;
}


void solve()
{
	p[1]=0;
	build(1);
	dfs(1);
	int ans=9999999,ans_n;
	for(int i=1;i<=n;i++)
	{
		if(banlance[i]<ans)
		{
			ans=banlance[i];
			ans_n=i;
		}
	}
	printf("%d %d\n",ans_n,banlance[ans_n]);
	
}

int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		init();
		solve();
	}
	return 0;
}