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）

 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #include<vector>
 #include<queue>
 using namespace std;
 #define fi first
 #define se second
 #define mp make_pair
 #define pb push_back
 typedef long long ll;
 typedef unsigned long long ull;
 typedef pair<int,int> pii;
 struct E
 {
     int to,nxt;
 }e[];
 int f1[],ne;
 int rt;
 bool inq[];
 int dep[],d[],mlen[],sz[],mson[];
 void dfs1(int u,int fa)
 {
     dep[u]=dep[fa]+;
     for(int k=f1[u];k;k=e[k].nxt)
         if(e[k].to!=fa)
         {
             dfs1(e[k].to,u);
             mlen[u]=max(mlen[u],mlen[e[k].to]);
             sz[u]+=sz[e[k].to];
         }
     mlen[u]++;
     sz[u]++;
 }
 void dfs2(int u,int fa)
 {
     for(int k=f1[u];k;k=e[k].nxt)
         if(e[k].to!=fa)
             if(!mson[u]||mlen[e[k].to]>mlen[mson[u]])
                 mson[u]=e[k].to;
     d[u]=;
     if(u==mson[fa])    d[u]+=d[fa];
     for(int k=f1[u];k;k=e[k].nxt)
         if(e[k].to!=fa)
             dfs2(e[k].to,u);
 }
 queue<int> q;
 int n,nn;
 int tt[];
 int main()
 {
     int i,x,y,u;
     scanf("%d",&n);nn=n;
     for(i=;i<n;i++)
     {
         scanf("%d%d",&x,&y);
         e[++ne].to=y;e[ne].nxt=f1[x];f1[x]=ne;
         e[++ne].to=x;e[ne].nxt=f1[y];f1[y]=ne;
     }
     q.push();inq[]=;
     while(!q.empty())
     {
         u=q.front();q.pop();
         rt=u;
         for(int k=f1[u];k;k=e[k].nxt)
             if(!inq[e[k].to])
             {
                 inq[e[k].to]=;
                 q.push(e[k].to);
             }
     }
     dfs1(rt,);
     dfs2(rt,);
     printf("%d ",);nn--;
     /*
     for(i=1;i<=n;i++)
         printf("t%d %d %d\n",i,sz[i],d[i]);
     */
     for(i=;i<=n;i++)
         if(sz[i]==)
             tt[++tt[]]=d[i];
     sort(tt+,tt+tt[]+);
     int sum=;
     for(i=tt[];i>=&&nn;i--)
     {
         sum+=tt[i];
         printf("%d ",sum);
         nn--;
     }
     //puts("test");
     for(;nn;nn--)    printf("%d ",sum);
     return ;
 }