树的重心 poj3701 模板题

树的重心也叫树的质心。找到一个点,其所有的子树中最大的子树节点数最少,那么这个点就是这棵树的重心,删去重心后,生成的多棵树尽可能平衡。

思路:dfs深搜+回溯

poj3701  (有多个重心)

AC代码:

#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
#define LL long long
const int MOD=1e9+7;
const int inf=0x3f3f3f3f;
const LL inff=0x3f3f3f3f3f3f3f3f;
const LL MAX_N=10005;
const LL MAX_M=100005;
#define MEF(x) memset(x,-1,sizeof(x))
#define ME0(x) memset(x,0,sizeof(x))
#define MEI(x) memset(x,inf,sizeof(x))
struct Edge
{
    int v,next;
}edge[100005];
int first[50005],cnt=0;
void init()
{
    cnt=0;
    MEF(first);
}
void add(int u,int v)
{
    edge[cnt].v=v;
    edge[cnt].next=first[u];
    first[u]=cnt++;
}
int num[50005],al[50005],l,n,mint;//num[i]表示i为根,子树节点数的个数
                                //tamp  子树节点个数的最大值,mint tamp的最小值
void dfs(int u,int fa)
{
    num[u]=1;
    int tamp=-inf;
    for(int i=first[u];i!=-1;i=edge[i].next)
    {
        int lv=edge[i].v;
        if(lv==fa)
        {
            continue;
        }
        dfs(lv,u);
        num[u]+=num[lv];
        if(tamptamp)
    {
        mint=tamp;
        l=0;
        al[l++]=u;
    }
    else if(tamp==mint)
    {
        al[l++]=u;
    }
    return ;
}
int main()
{
    scanf("%d",&n);
    init();
    for(int n1=1,a,b;n1<=n-1;n1++)
    {
        scanf("%d%d",&a,&b);
        add(a,b);
        add(b,a);
    }
    mint=inf;
    l=0;
    ME0(num);
    dfs(1,-1);
    sort(al,al+l);
    for(int i=0;i

 

你可能感兴趣的:(树的重心 poj3701 模板题)