[题解/模板]luogu_P3942_（树上覆盖问题

Posted 2022-12-04 superminivan

抄题解

把点按深度排序，用near数组记录到每个点最近的关键点的距离，每次取出一个点更新一下near数组，如果不能被覆盖就在它的k级祖先建立关键点，并更新所有k级祖先的k级祖先的near数组

#include<bits/stdc++.h>
using namespace std;
const int maxn=100009;
int n,k,t,ans;
struct edge
    int v,nxt;
e[maxn<<1];
int head[maxn],cnt;
inline void add(int u,int v)
    e[++cnt].v=v;e[cnt].nxt=head[u];head[u]=cnt;

struct node
    int dp,id;
    bool operator <(const node&t)const
        return dp>t.dp;
    
p[maxn];
int fa[maxn],nr[maxn];
void pre(int x,int f)
    p[x].dp=p[f].dp+1;fa[x]=f;
    for(int i=head[x];i;i=e[i].nxt)
        if(e[i].v==f)continue;
        pre(e[i].v,x);
    

int main()
    scanf("%d%d%d",&n,&k,&t);
    for(int i=0;i<=n;i++)
        nr[i]=maxn;p[i].id=i;
    
    for(int i=1,u,v;i<n;i++)
        scanf("%d%d",&u,&v);
        add(u,v);add(v,u);
    
    pre(1,0);
    sort(p+1,p+1+n);
    for(int i=1;i<=n;i++)
        int x=p[i].id,y=x;
        for(int j=1;j<=k;j++)
            nr[x]=min(nr[x],nr[fa[y]]+j),y=fa[y];
        
        if(nr[x]>k)
            nr[y]=0;ans++;
            for(int j=1;j<=k;j++)
                nr[fa[y]]=min(nr[fa[y]],j);y=fa[y];
            
        
    
    printf("%d",ans);