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

E.小G有一个大树

 

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 32768K，其他语言65536K
64bit IO Format: %lld

链接：https://www.nowcoder.com/acm/contest/74/E
来源：牛客网

题目描述

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

输入描述:

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

输出描述:

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

示例1

输入

3
1 2
1 3

输出

1 1

这个题是树的重心，他们直接找的板子过的，我也没仔细看。。。

代码:

#include <cstdio>
#include <iostream>
#include <fstream>
#include <cstring>
#include <string>
#include <vector>
#define OP(s) cout<<#s<<"="<<s<<" ";
#define PP(s) cout<<#s<<"="<<s<<endl;
using namespace std;
int n;
vector <int> adj[20010];
int son[20010];
bool vd[20010];
int ans,asize = 1<<29;
void DFS(int s){
    vd[s] = 1;
    son[s] = 0;
    int blance = 0;
    int size = adj[s].size();
    for (int j = 0;j < size;j++){
        int u = adj[s][j];
        if (vd[u]) continue;
        DFS(u);
        son[s] += son[u]+1;
        blance = max(blance,son[u]+1);
    }
    blance = max(blance,n - son[s] - 1);
    if (blance < asize || blance == asize && s < ans)
        ans = s,asize = blance;
}
int main(){
    cin>>n;
    for (int i = 1;i <= n;i++) adj[i].clear();
    for (int i = 1;i <= n-1;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        adj[u].push_back(v);
        adj[v].push_back(u);
    }
    memset(vd,0,sizeof(vd));
    asize = 1<<29;
    DFS(1);
    cout<<ans<<" "<<asize<<endl;
    return 0;
}