2018年全国多校算法寒假训练营练习比赛（第二场）-E小G有一个大树【树的重心】

模板题

题目描述 

小G想要把自己家院子里的橘子树搬到家门口（QAQ。。就当小G是大力水手吧）
可是小G是个平衡性灰常灰常差的人，他想找到一个这个橘子树的平衡点。
怎么描述这棵树呢。。。就把它看成由一个个节点构成的树吧。结点数就
代表树重。

输入描述:

多组数据输入输出，
第一行包含一个整数n（3<=n<=1000）代表树的结点的个数
以下n-1行描述（1-n）节点间的连接关系。

输出描述:

输出两个个整数 x，num 分别代表树的平衡点，和删除平衡点后最大子树的结点数（如果结点数相同输出编号小的）。

示例1

输入

3
1 2
1 3

输出

1 1

代码：

#include<stdio.h>
#include<vector>
#include<algorithm>
#include<string.h>
using namespace std;
 
#define N 1005
#define inf 0x3f3f3f3f
vector<int>g[N];
 
int Max[N];
int Num[N];
int n;
int ans;
int dp_Num(int u,int pre)
{
    int k=g[u].size();
    Num[u]=1;
    for(int i=0; i<k; i++)
    {
        if(g[u][i]==pre)continue;
        Num[u]+=dp_Num(g[u][i],u);
    }
    return Num[u];
}
 
void dp_Max(int u,int pre)
{
    int k=g[u].size();
    int Mod = 0;
    for(int i=0; i<k; i++)
    {
        if(g[u][i]==pre)continue;
        dp_Num(g[u][i],u);
        dp_Max(g[u][i],u);
        Mod += Num[g[u][i]];
        Max[u]=max(Max[u],Num[g[u][i]]);
    }
    Max[u]=max(Max[u],n-Mod-1);
    return;
}
 
int main()
{
    while(scanf("%d",&n)==1)
    {
        int u,v;
        memset(Max,0,sizeof(Max));
        memset(Num,0,sizeof(Num));
        for(int i=1; i<n; i++)
        {
            scanf("%d%d",&u,&v);
            g[u].push_back(v);
            g[v].push_back(u);
        }
        int p=0;
        Max[0]=inf;
        dp_Max(1,0);
        for(int i=1; i<=n; i++)
        {
            if(Max[p]>Max[i])
            {
                p=i;
                 ans=Max[p];
            }
        }
        printf("%d %d\n",p,ans);
        for(int i=0; i<N; i++)g[i].clear();
    }
}