POJ1655 Balancing Act

树的重心

题目传送门

题目大意：给你t组数据，每一组数据为一棵树，求树的重心。

基础题，关键要知道树的重心的定义。
直接dfs暴力跑过。

贴上代码（刚开始RE我以为数组开小了，这里没改）：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
struct tree{
    int next;
    int to;
};
int t,n,m,node,ans;
tree a[100005];
int sum[50005];
int h[50005];
bool f[50005];
void read(int x,int y){
    m++;
    a[m].next=h[x];
    a[m].to=y;
    h[x]=m;
}
void dfs(int x){
    f[x]=true;
    int ma=0;
    for (int i=h[x];i;i=a[i].next){
        if (!f[a[i].to]){
            dfs(a[i].to);
            sum[x]+=sum[a[i].to]+1;
            ma=max(ma,sum[a[i].to]+1);
        }
    }
    ma=max(ma,n-1-sum[x]);
    if (ma<ans||ma==ans&&x<node){
        ans=ma;
        node=x;
    }
}
int main(){
    scanf("%d",&t);
    while (t--){
        scanf("%d",&n);
        memset(h,0,sizeof(h));
        memset(sum,0,sizeof(sum));
        memset(f,false,sizeof(f));
        m=0;
        for (int i=1;i<=n-1;i++){
            int x,y;
            scanf("%d%d",&x,&y);
            read(x,y);
            read(y,x);
        }
        ans=0x7fffffff;
        dfs(1);
        printf("%d %d\n",node,ans);
    }
    return 0;
}