POJ 3107 Godfather (求树的重心）

最新推荐文章于 2023-09-26 23:36:32 发布

最新推荐文章于 2023-09-26 23:36:32 发布 · 1.7k 阅读

本文介绍了一种利用树的重心概念解决特定问题的方法，并通过一个实际的编程实例展示了如何求解树的重心，包括关键的数据结构定义、递归求解子树大小及重心的具体算法。

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

看到这题我很惊讶，求删掉某点后子树节点最多的最小。这不就是求树的重心吗？而求重心在之前的POJ 1741已经做过了（而且在里面只是一小步）。直接把代码复制一下。。。

具体代码里面也注释了。

有一个trick点，别忘了重心要排序后输出，因此wa了一次

【代码】

/* ***********************************************
Author        :angon

************************************************ */
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <stack>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
#define showtime fprintf(stderr,"time = %.15f\n",clock() / (double)CLOCKS_PER_SEC)
#define lld %I64d
#define REP(i,k,n) for(int i=k;i<n;i++)
#define REPP(i,k,n) for(int i=k;i<=n;i++)
#define scan(d) scanf("%d",&d)
#define scanl(d) scanf("%I64d",&d)
#define scann(n,m) scanf("%d%d",&n,&m)
#define scannl(n,m) scanf("%I64d%I64d",&n,&m)
#define mst(a,k)  memset(a,k,sizeof(a))
#define LL long long
#define N 50005
#define mod 1000000007
inline int read(){int s=0;char ch=getchar();for(; ch<'0'||ch>'9'; ch=getchar());for(; ch>='0'&&ch<='9'; ch=getchar())s=s*10+ch-'0';return s;}

struct Edge
{
    int to,next;
}edge[N*2];
int head[N],tot;

void addedge(int u,int v)
{
    edge[tot].to = v;
    edge[tot].next = head[u];
    head[u] = tot++;
}
int sonsize[N],sonmx[N];//子树大小，子树最大 大小
void dfssize(int u,int fa)  //求子树大小
{
    sonsize[u]=1; sonmx[u]=0;
    for(int i=head[u];~i;i=edge[i].next)
    {
        int v=edge[i].to;
        if(v==fa) continue;
        dfssize(v,u);
        sonsize[u] += sonsize[v];  //  DP思想
        sonmx[u] = max(sonmx[u],sonsize[v]);  //保留最大
    }
}
int root[N];
int num,minc;
void dfsroot(int r,int u,int fa) //求根为r的树的重心
{
    if(sonsize[r]-sonsize[u] > sonmx[u]) sonmx[u] = sonsize[r]-sonsize[u]; //u节点向上的子树
    if(sonmx[u]<minc) //子树最大的最小即为重心
    {
        minc=sonmx[u];
        num=0;
        root[num++]=u;
    }
    else if(sonmx[u]==minc)
    {
       root[num++]=u;
    }
    for(int i=head[u];~i;i=edge[i].next)
    {
        int v=edge[i].to;
        if(v==fa) continue;
        dfsroot(r,v,u);
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int n;
    while(~scan(n))
    {
        mst(head,-1); tot=0;
        REP(i,1,n)
        {
            int u,v;
            scann(u,v);
            addedge(u,v);
            addedge(v,u);
        }
        dfssize(1,-1);
        minc = 1<<30;
        dfsroot(1,1,-1);
        sort(root,root+num);
        for(int i=0;i<num;i++)
            printf("%d%c",root[i],i<num-1?' ':'\n');
    }


    return 0;
}