POJ-1655-树的重心

树的重心模板题

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <map>
#include <queue>
#include <functional>
#include <vector>
#include <stack>
#include <set>
#include <bitset>
using namespace std;
typedef  long long ll;
const int maxn=1e5+50;
const int inf=0x3f3f3f3f;
const int MOD=1e9+7;
const int HASH=131;

int head[maxn];//前向星
int tot;//前向星
int siz;//所有子树最大结点数量
int ans;//重心编号
int sz[maxn];//每颗子树最大结点数量
int n;

struct node
{
    int to,next;
}E[maxn*2];

void adde(int x,int y)
{
    E[tot].to=y;
    E[tot].next=head[x];
    head[x]=tot++;
}

void add(int x,int y)
{
    adde(x,y);
    adde(y,x);
}

void dfs(int u,int father)
{
    sz[u]=1;
    int num=0;//num存大小
    for(int i=head[u];~i;i=E[i].next)
    {
        int j=E[i].to;
        if(j!=father)//防止出现环
        {
            dfs(j,u);
            sz[u]+=sz[j];//回溯，更新
            num=max(num,sz[j]);//每遍历一颗子树判断一次
        }
    }
    num=max(num,n-sz[u]);//n为总结点数量,n-sz[u]为向上的节点个数,
                         //比较完之后便是当前节点的最大子树节点
    if(num<siz||(num==siz&&u<ans))
    {
        ans=u;
        siz=num;
    }
}

void init()//初始化
{
    memset(sz,0,sizeof(sz));
    memset(head,-1,sizeof(head));
    tot=0;
    siz=inf;
    ans=-1;
}

int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        cin>>n;
        init();
        for(int i=1;i<n;i++)
        {
            int x,y;
            cin>>x>>y;
            add(x,y);
        }
        dfs(1,0);
        printf("%d %d\n",ans,siz);
    }
    return 0;
}