[CF1029E] Tree with Small Distances

题目传送门

洛谷 题面翻译

树上贪心。

先预处理出距离。我就用dijkstra皮一下QAQ

显然每个距离大于2的点都需要考虑。

我们贪心处理，优先处理距离最大的。

每次连接距离最大的点的父节点和一号节点。

然后打标记就行了。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;

int n,ans;
int hd[200005],to[400005],nx[400005],ec;
int dis[200005],f[200005];
bool v[200005];

void edge(int af,int at)
{
    to[++ec]=at;
    nx[ec]=hd[af];
    hd[af]=ec;
}

struct data
{
    int p,d;
    friend bool operator < (data q,data w)
    {
        return q.d>w.d;
    }
};

priority_queue<data>qq;

void dijkstra()
{
    memset(dis,0x3f,sizeof(dis));
    dis[1]=0;
    qq.push((data){1,0});
    while(!qq.empty())
    {
        data nw=qq.top();
        qq.pop();
        if(v[nw.p])continue;
        v[nw.p]=1;
        for(int i=hd[nw.p];i;i=nx[i])
        {
            if((!v[to[i]])&&dis[to[i]]>nw.d+1)
            {
                dis[to[i]]=nw.d+1;
                f[to[i]]=nw.p;
                qq.push((data){to[i],dis[to[i]]});
            }
        }
    }
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int ff,tt;
        scanf("%d%d",&ff,&tt);
        edge(ff,tt);
        edge(tt,ff);
    }
    dijkstra();
    memset(v,0,sizeof(v));
    for(int i=1;i<=n;i++)
    {
        if(dis[i]>2)qq.push((data){i,-dis[i]});
        else v[i]=1;
    }
    while(!qq.empty())
    {
        data nw=qq.top();
        qq.pop();
        if(v[nw.p])continue;
        int fa=f[nw.p];
        v[fa]=1;
        ans++;
        for(int i=hd[fa];i;i=nx[i])
        {
            v[to[i]]=1;
        }
    }
    printf("%d",ans);
    return 0;
}