题目:就是找树的重心,可能存在多个,按编号从小到大输出。一个水题,但是好几个罚时....不能用vector存边,会TLE.
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=5e4+100;
struct node
{
int v;
int next;
}edge[2*maxn];
int first[maxn],tot;
void addedge(int u,int v)
{
edge[tot].v=v;
edge[tot].next=first[u];
first[u]=tot++;
}
int hvy,n,m,siz[maxn],ma[maxn];
int S; //树的大小
int num;//重心的数量
void dfs_hvy(int u,int fa)//找重心
{
siz[u]=1;
ma[u]=0;
for(int i=first[u];i!=-1;i=edge[i].next)
{
int v=edge[i].v;
if(v==fa) continue;
dfs_hvy(v,u);
siz[u]+=siz[v];
ma[u]=max(ma[u],siz[v]);
}
ma[u]=max(ma[u],S-siz[u]);
if(!hvy||ma[u]<ma[hvy])// 找重心,以重心为根的树的最大子树最小
{
hvy=u;
num=1;
}
else if(hvy&&ma[u]==ma[hvy])
num++;
}
int main()
{
while(~scanf("%d",&n))
{
tot=0;
for(int i=1;i<=n;i++)
first[i]=-1;
for(int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
addedge(x,y);
addedge(y,x);
}
S=n;hvy=0;
dfs_hvy(1,-1);//找重心
int k=0;
for(int i=1;i<=n;i++)
if(ma[i]==ma[hvy])
{
k++;
if(k<num)
printf("%d ",i);
else printf("%d\n",i);
}
}
return 0;
}