树的重心（POJ1655）

随手wa

于 2020-05-26 17:25:44 发布

阅读量273

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.csdn.net/zhlnb/article/details/106360606

本文深入探讨了树的重心算法，介绍了如何通过寻找树中所有最大子树的最小大小来确定树的重心，提供了完整的C++代码实现，并通过具体实例展示了算法的应用过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

树的重心：设size[x]为x点的所有最大子树的大小，树的重心为min（size[x]）

//#pragma GCC optimize(3,"Ofast","inline")
//ios::sync_with_stdio(false);
//#include<unordered_map>
//#include<unordered_set>
#include<cstdio>
#include<iostream>
#include<cmath>
#include<functional>
#include<cstring>
#include<string>
#include<cstdlib>
#include<queue>
#include<map>
#include<algorithm>
#include<set>
#include<stack>
#include<vector>
#include<sstream>
using namespace std;
typedef long long ll;
const int INF=0x3f3f3f3f;
const int maxn= 2e4+100;
const ll mod=998244353;
struct edge
{
    int from,to,next;
}e[2*maxn];
int num=0;
int t,n;
int head[maxn];
void add(int from,int to)
{
    e[num].from=from;
    e[num].to=to;
    e[num].next=head[from];
    head[from]=num++;
}
int sizen[maxn];
int vis[maxn];
int ans=INF;
int ansn;
int dfs(int x)
{
    vis[x]=1;
    int sum=0;
    for(int i=head[x];i!=-1;i=e[i].next)
    {
        int j=e[i].to;
        if(vis[e[i].to])continue;
        int s=dfs(j);
        sizen[x]=max(sizen[x],s);//选最大的子树
        sum+=s;
    }
    sizen[x]=max(sizen[x],n-sum-1);//比较上方子树
    if(sizen[x]<ans)//答案在这里
    {
        ans=sizen[x];
        ansn=x;
    }
    return sum+1;
}
int main()
{
    ios::sync_with_stdio(false);
    int x,y;
    cin>>t;
    while(t--)
    {
        memset(vis,0,sizeof(vis));
        memset(head,-1,sizeof head);
        memset(sizen,0,sizeof sizen);
        ans=INF;
        num=0;
        cin>>n;
        for(int i=0;i<n-1;i++)
        {
            cin>>x>>y;
            add(x,y);
            add(y,x);
        }
        dfs(1);
        cout<<ansn<<" "<<ans<<endl;
    }
    return 0;
}