Maximum Control (medium) Codeforces - 958B2
https://codeforces.com/contest/958/problem/B2
题解：https://www.cnblogs.com/Cool-Angel/p/8862649.html
upd2018-11-01:
修了一个bug（第60行加入inq[1]=1）
1 #include<cstdio> 2 #include<algorithm> 3 #include<cstring> 4 #include<vector> 5 #include<queue> 6 using namespace std; 7 #define fi first 8 #define se second 9 #define mp make_pair 10 #define pb push_back 11 typedef long long ll; 12 typedef unsigned long long ull; 13 typedef pair<int,int> pii; 14 struct E 15 { 16 int to,nxt; 17 }e[200100]; 18 int f1[100100],ne; 19 int rt; 20 bool inq[101000]; 21 int dep[101000],d[101000],mlen[100100],sz[101000],mson[101000]; 22 void dfs1(int u,int fa) 23 { 24 dep[u]=dep[fa]+1; 25 for(int k=f1[u];k;k=e[k].nxt) 26 if(e[k].to!=fa) 27 { 28 dfs1(e[k].to,u); 29 mlen[u]=max(mlen[u],mlen[e[k].to]); 30 sz[u]+=sz[e[k].to]; 31 } 32 mlen[u]++; 33 sz[u]++; 34 } 35 void dfs2(int u,int fa) 36 { 37 for(int k=f1[u];k;k=e[k].nxt) 38 if(e[k].to!=fa) 39 if(!mson[u]||mlen[e[k].to]>mlen[mson[u]]) 40 mson[u]=e[k].to; 41 d[u]=1; 42 if(u==mson[fa]) d[u]+=d[fa]; 43 for(int k=f1[u];k;k=e[k].nxt) 44 if(e[k].to!=fa) 45 dfs2(e[k].to,u); 46 } 47 queue<int> q; 48 int n,nn; 49 int tt[100100]; 50 int main() 51 { 52 int i,x,y,u; 53 scanf("%d",&n);nn=n; 54 for(i=1;i<n;i++) 55 { 56 scanf("%d%d",&x,&y); 57 e[++ne].to=y;e[ne].nxt=f1[x];f1[x]=ne; 58 e[++ne].to=x;e[ne].nxt=f1[y];f1[y]=ne; 59 } 60 q.push(1);inq[1]=1; 61 while(!q.empty()) 62 { 63 u=q.front();q.pop(); 64 rt=u; 65 for(int k=f1[u];k;k=e[k].nxt) 66 if(!inq[e[k].to]) 67 { 68 inq[e[k].to]=1; 69 q.push(e[k].to); 70 } 71 } 72 dfs1(rt,0); 73 dfs2(rt,0); 74 printf("%d ",1);nn--; 75 /* 76 for(i=1;i<=n;i++) 77 printf("t%d %d %d\n",i,sz[i],d[i]); 78 */ 79 for(i=1;i<=n;i++) 80 if(sz[i]==1) 81 tt[++tt[0]]=d[i]; 82 sort(tt+1,tt+tt[0]+1); 83 int sum=0; 84 for(i=tt[0];i>=1&&nn;i--) 85 { 86 sum+=tt[i]; 87 printf("%d ",sum); 88 nn--; 89 } 90 //puts("test"); 91 for(;nn;nn--) printf("%d ",sum); 92 return 0; 93 }
