【求树的重心】POJ - 3107 Godfather

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct node
{
int to, next;
};
node e[100000];
//e,head是前向星存图用的数组。n个结点，ans[]用来记录最后结果,vis[]用来标记那些点跑过，Size[]用来记录它的子数的结点数，n-size[]对应的就是父亲那边的结点数。
int head[100000], n, ans[50005], vis[50005], Size[50005], m, top;
void dfs(int u)//dfs遍历所有结点
{
int v;
vis[u] = 1;//标记已经走过
Size[u] = 1;//走到的结点初始化为1
int t = 0;//用来存储孩子最大的Size
for(int i = head[u]; ~i; i = e[i].next)
{
v = e[i].to;
if(!vis[v])//没有走过的结点
{
dfs(v);
Size[u] += Size[v];//回潮的时候，更新父亲结点的Size
t = max(t, Size[v]);//更新t的大小
}
}
t = max(t, n - Size[u]);//求孩子Size，和本身Size最大的
if(t < m)//更新树的重心所有子数，最大子树的结点，存入结果
{
top = 0;
m = t;
ans[top++] = u;
}
else if(t == m)//多个重心
{
ans[top++] = u;
}
}
int main()
{
int u, v;
while(~scanf("%d", &n))
{
m = n;
memset(head, -1, sizeof(head));//初始化
memset(vis, 0, sizeof(vis));
memset(Size, 0, sizeof(Size));
int cnt = 0;
for(int i = 0; i < n - 1; i++)//前向星存图
{
scanf("%d %d", &u, &v);
e[cnt].to = v;
e[cnt].next = head[u];
head[u] = cnt++;
e[cnt].to = u;
e[cnt].next = head[v];
head[v] = cnt++;
}
dfs(1);//求树的重心
sort(ans, ans + top);//排序
for(int i = 0; i < top; i++)//输出
{
if(i) printf(" ");
printf("%d", ans[i]);
}
printf("\n");
}
return 0;
}