poj 1655 Balancing Act (树形dp)

对于每个节点，它的balance要么来自于父亲节点，要么来自于儿子节点。
定义dp[u]为节点u的balance，sum[u]为以u为根的子树的节点个数，n为总共的节点个数。
如果来自于父亲节点，则dp[u]=n-sum[u];
如果来自于儿子节点，则dp[u]=sum[v];

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<vector>
#define N 20005
using namespace std;

vector<int>edge[N];
int sum[N],dp[N],n;

void dfs(int u,int fa)
{
    int i,len=edge[u].size(),v;
    sum[u]=1;
    dp[u]=0;
    for(i=0;i<len;i++){
        v=edge[u][i];
        if(v==fa) continue;
        dfs(v,u);
        dp[u]=max(dp[u],sum[v]);
        sum[u]+=sum[v];
    }
    dp[u]=max(dp[u],n-sum[u]);
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--){
        int i;
        scanf("%d",&n);
        for(i=0;i<=n;i++) edge[i].clear();
        for(i=1;i<n;i++){
            int a,b;
            scanf("%d%d",&a,&b);
            edge[a].push_back(b);
            edge[b].push_back(a);
        }
        dfs(1,0);
        int ans=99999999,si;
        for(i=1;i<=n;i++){
            if(dp[i]<ans){
                ans=dp[i];
                si=i;
            }
        }
        printf("%d %d\n",si,ans);
    }
    return 0;
}