【AGC004D】Teleporter【结论题】

先把1号节点连向自己，然后从深往浅贪心，每隔k的深度连一条边到1。

#include<bits/stdc++.h>
using namespace std;
const int N=200005;
int n,k,ans,cnt,a[N],dep[N],head[N],to[N*2],nxt[N*2];
void adde(int u,int v){
    to[++cnt]=v;
    nxt[cnt]=head[u];
    head[u]=cnt;
}
void dfs(int pre,int u){
    int v;
    for(int i=head[u];i;i=nxt[i]){
        v=to[i];
        if(v!=pre){
            dfs(u,v);
            dep[u]=max(dep[u],dep[v]+1);
        }
    }
    if(dep[u]==k-1&&a[u]!=1){
        ans++;
        dep[u]=-1;
    }
}
int main(){
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    if(a[1]!=1){
        ans++;
        a[1]=1;
    }
    for(int i=2;i<=n;i++){
        adde(i,a[i]);
        adde(a[i],i);
    }
    dfs(0,1);
    printf("%d\n",ans);
    return 0;
}