和poj1655的方法完全一样,但是这道题的n的范围大了,用vector存图会TLE(poj没有O2编译优化),所以改用前向星来存图就可以了。、
有关树的重心,看这里:poj1655
这里解释一下前向星存图的方法:
其实就是用静态链表来实现邻接链表,这样可以避免使用指针。
head[i]数组来记录每个节点的第一条边;每条边用结构体e[i]来存,e[i].v表示这条边指向的点,e[i].next表示这条边连向的下一条边。
它的巧妙之处在于每次插入到链表的首部而不是尾部,这样就避免了对链表的遍历。同一起点的各条边在邻接表中的顺序和读入顺序正好相反。
贴个模板:
struct node { int v,next; }e[m];//m是总边数 int head[n],cnt;//n是总节点数,cnt记录边数 void init() { memset(head,-1,sizeof(head)); cnt=0; } void add(int u,int v) { e[cnt].v=v; e[cnt].next=head[u]; head[u]=cnt++; }
本题AC代码:
#include<iostream> #include<cstdio> #include<cstdlib> #include<cstring> #include<cmath> #include<map> #include<set> #include<vector> #include<algorithm> #include<stack> #include<queue> using namespace std; #define INF 100000000 #define eps 1e-8 #define pii pair<int,int> #define LL long long int struct node { int v,next; }e[100010]; int n,a,b,head[50005],mi=1; int num[50005],bal[50005],cnt=1; void add(int aa,int bb); int dfs1(int x,int fa); void dfs2(int x,int fa); int main() { //freopen("in1.txt","r",stdin); //freopen("out.txt","w",stdout); scanf("%d",&n); memset(head,-1,sizeof(int)*(n+1)); for(int i=1;i<=n;i++) num[i]=1; for(int i=1;i<=n-1;i++) { scanf("%d%d",&a,&b); add(a,b); add(b,a); } dfs1(1,-1); dfs2(1,-1); for(int i=1;i<=n;i++) { if(bal[i]<bal[mi]) { mi=i; } } printf("%d",mi); for(int i=mi+1;i<=n;i++) { if(bal[i]==bal[mi]) { printf(" %d",i); } } printf(" "); //fclose(stdin); //fclose(stdout); return 0; } void add(int aa,int bb) { e[cnt].v=bb; e[cnt].next=head[aa]; head[aa]=cnt++; } int dfs1(int x,int fa) { for(int i=head[x];i!=-1;i=e[i].next) { if(e[i].v==fa) continue; else { num[x]+=dfs1(e[i].v,x); } } return num[x]; } void dfs2(int x,int fa) { for(int i=head[x];i!=-1;i=e[i].next) { if(e[i].v==fa) { bal[x]=max(bal[x],n-num[x]); } else { bal[x]=max(bal[x],num[e[i].v]); dfs2(e[i].v,x); } } }