SGU 134 树形DP

SGU 134
题目链接：
http://acm.sgu.ru/problem.php?contest=0&problem=134
题意：
求一个数的重心，即把这个点去掉后，剩余几个连通分量中点最多的那个连通分量点最少。
思路：
树形DP一下就可以，因为把这点去点后，存在两种情况：
1）子树形成连通分量
2）除了它和它子树的其他点形成一个连通分量。、
然后求出这几个值得最大值就可以。
然后对每个点的这个最大值，用来更新一下ans。
SGU上不能用EOF。
源码：

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <iostream>
#include <string>
#include <vector>
using namespace std;
const int MAXN = 16000 + 5;
int num[MAXN];
int dp[MAXN];
int ans[MAXN], cnt;
int vis[MAXN];
vector<int>lin[MAXN];
int n;
int mmax;
void dfs(int u)
{
    vis[u] = 1;
    num[u] = 1;
    dp[u] = 0;
    for(int i = 0 ; i < (int)lin[u].size() ; i++){
        int v = lin[u][i];
        if(vis[v] == 0){dfs(v); num[u] += num[v]; dp[u] = max(dp[u], num[v]);}
    }
    int temp = max(dp[u], n - num[u]);
    if(mmax > temp){
        mmax = temp;
        cnt = 0;
        ans[cnt++] = u;
    }
    else if(mmax == temp)   ans[cnt++] = u;
}
int main()
{
    scanf("%d", &n);
        for(int i = 1 ; i <= n ; i++)
            lin[i].clear(), num[i] = 0, vis[i] = 0;
        int u, v;
        for(int i = 0 ; i < n - 1 ; i++)
            scanf("%d%d", &u, &v), lin[u].push_back(v), lin[v].push_back(u);
        cnt = 0;
        mmax = n;
        dfs(1);
        sort(ans, ans + cnt);
        printf("%d %d\n", mmax, cnt);
        for(int i = 0 ; i < cnt ; i++){
            printf("%d", ans[i]);
            if(i == cnt - 1)    printf("\n");
            else    printf(" ");
        }
    return 0;
}