CF1029E Tree with Small Distances (贪心)

题目大意：给你一棵边权为1的树，让你加入一些边，使得根节点(1号节点)到其他节点的最短距离不大于2

并没有想到贪心......

正解的贪心思路是这样的

用一个堆维护当前距离最远的点，然后把根节点和它的父节点连起来

这样，父节点周围一圈的节点到根的距离都不大于2，把这些节点都从堆里删除

实际操作的时候，并不需要删除它们，只需要把它们标记一下，等它们弹出堆的时候，continue即可

不断按照这个思路贪心，直到所有节点到根的距离都不大于2即可

 #include <set>
 #include <queue>
 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #define N 200010
 #define ll long long
 using namespace std;
 //re
 int gint()
 {
     int ret=,fh=;char c=getchar();
     while(c<''||c>''){if(c=='-')fh=-;c=getchar();}
     while(c>=''&c<=''){ret=ret*+c-'';c=getchar();}
     return ret*fh;
 }
 int n,m,K,ma,cte,num;
 int head[N],vis[N],use[N],dis[N],fa[N];
 struct Edge{int to,nxt,val;}edge[N*];
 void ae(int u,int v){
     cte++;edge[cte].to=v;
     edge[cte].nxt=head[u],head[u]=cte;}
 struct node{
     int id,d;
     friend bool operator<(const node &s1,const node &s2)
     {return s1.d<s2.d;}
     node(int id,int d):id(id),d(d){} node(){}
 };
 void bfs()
 {
     queue<int>q;q.push();
     while(!q.empty())
     {
         int u=q.front();q.pop();
         for(int j=head[u];j;j=edge[j].nxt){
             int v=edge[j].to;
             if(v==fa[u]) continue;
             fa[v]=u,dis[v]=dis[u]+;
             q.push(v);
         }
     }
 }
 int solve()
 {
     priority_queue<node>q;
     for(int i=;i<=n;i++)
         if(dis[i]>) q.push(node(i,dis[i]));
     int ans=;
     while(!q.empty())
     {
         node k=q.top();q.pop();
         int x=k.id,y=fa[x];
         if(vis[x]) continue; vis[x]=;
         dis[y]=,vis[y]=,ans++;
         for(int j=head[y];j;j=edge[j].nxt){
             int v=edge[j].to;
             dis[v]=min(dis[v],),vis[v]=;
             q.push(node(v,dis[v]));
         }
     }return ans;
 }

 int main()
 {
     scanf("%d",&n);
     int x,y,z,fx;
     for(int i=;i<n;i++)
         x=gint(),y=gint(),ae(x,y),ae(y,x);
     bfs();
     printf("%d\n",solve());
     return ;
 }