POJ 1655 Balancing Act

实际上，这个题不难，状态很好写：balace[i]=max(size[j],N-size[i])，其中i是j的父节点，size[i]表示以节点i为根节点所构成子树的节点总数。

一遍dfs就可以搞定。题目里给的数据是告诉你每两个节点之间有一条边，而不是父子节点关系。。。

#include <iostream>
#include <cstdio>
#include <vector>
#define MAX 0xffffff
using namespace std;

vector <int> G[20001];
int balance[20001];
bool visit[20001];
int n,t;

int max(int a,int b)
{
    return a > b ? a : b;
}

int min(int a,int b)
{
    return a < b ? a : b;
}

int dfs(int v)
{
    visit[v] = true;
    int sum = 1,temp;
    for(vector<int>::size_type i = 0;i < G[v].size();i++)
    {
        int u = G[v][i];
        if(visit[u])
            continue;

        temp = dfs(u);
        sum += temp;
        balance[v] = max(balance[v],temp);
    }
    balance[v] = max(balance[v],n - sum);
    return sum;
}
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        int i,p,q;
        scanf("%d",&n);
        for(i = 1;i <= n;i++)
        {
            G[i].clear();
            balance[i] = -1;
            visit[i] = false;
        }
        for(i = 1;i < n;i++)
        {
            scanf("%d%d",&p,&q);
            G[p].push_back(q);
            G[q].push_back(p);
        }

        dfs(1);
        int flag,ans = MAX;
        for(i = 1;i <= n;i++)
        {
            if(balance[i] < ans)
            {
                ans = balance[i];
                flag = i;
            }
        }
        printf("%d %d\n",flag,ans);
    }
    return 0;
}