POJ 3107 Godfather (求树的重心）

看到这题我很惊讶，求删掉某点后子树节点最多的最小。这不就是求树的重心吗？而求重心在之前的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
,tot;

void addedge(int u,int v)
{
edge[tot].to = v;
edge[tot].next = head[u];
head[u] = tot++;
}
int sonsize
,sonmx
;//子树大小，子树最大 大小
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
;
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;
}