POJ 1655 Balancing Act (树的重心)

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

题意：

        求树的重心(最小序号)，以及去掉重心后子树节点最大是多少。（树的重心定义为:找到一个点,其所有的子树中最大的子树节点数最少,那么这个点就是这棵树的重心）

思路：

        dfs

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
const int N = 2e4 + 5;
int dp[N], n;
vector <int> G[N];

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

int main()
{
    int t;
    scanf("%d", &t);
    while(t--) {
        scanf("%d", &n);
        int u, v;
        for(int i = 1; i <= n; ++i) {
            G[i].clear();
        }
        for(int i = 1; i < n; ++i) {
            scanf("%d %d", &u, &v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        dfs(1, -1);
        int ans = n, pos = n;
        for(int i = n; i >= 1; --i) {
            if(dp[i] <= ans) {
                ans = dp[i];
                pos = i;
            }
        }
        printf("%d %d
", pos, ans);
    }
    return 0;
}