Poj 1655 【树的重心】

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

题解：

设sum[i]为以i为根结点的树上有多少个结点，maxsum[i]为去掉i结点导致的森林中最大的树有多大

一开始建树时是取小的点指向大的点，然后一直WA后来发现这样建树是有问题的，可能会有（1,3）（2,3）这种出现1->3,2->3导致1、2不联通的情况出现。

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<cmath>
#include<stdlib.h>
#include <string.h>
#include<queue>
#include<set>
#include<map>
#include<stack>
#include<time.h>
#include<sstream>
using namespace std;
#define MAX_N 20005
#define inf 0x7f7f7f7f
#define LL long long
#define ull unsigned long long
const LL INF = 1e18;
//const double Pi = acos(-1);
const int mod = 1e4+7;
typedef pair<int, int>P;

vector<int>G[MAX_N];
int sum[MAX_N];
int maxSum[MAX_N];
int n;
void dfs(int u, int pre)
{
    sum[u] = 1;
    maxSum[u] = 0;
    for(int i=0; i<G[u].size(); i++) {
        int v = G[u][i];
        if(v == pre)
            continue;
        dfs(v, u);
        sum[u] += sum[v];
        maxSum[u] = max(maxSum[u], sum[v]);
    }
    maxSum[u] = max(maxSum[u], n-sum[u]);
}
int main()
{
    int t;
    cin >> t;
    while(t--) {
        cin >> n;
        for(int i=1; i<=n; i++)
            G[i].clear();
        for(int i=1; i<n; i++) {
            int a, b;
            scanf("%d%d", &a, &b);
            G[a].push_back(b);
            G[b].push_back(a);
        }
        dfs(1, 0);
        int idx;
        int ans = inf;
        for(int i=1; i<=n; i++) {
            if(ans > maxSum[i]) {
                ans = maxSum[i];
                idx = i;
            }
        }
        printf("%d %d\n", idx, ans);
    }
}