#267 传送 [贪心]

#267 传送 [贪心]

题目传送门

题解

比赛的时候写了个10分贪心，然后结束前改了一下代码…然后…就彻底0分了……身败名裂QAQ

注意，题目里面有这样一句话：保证从任意城市出发，经过若干次传送，都能到达首都。我读掉了，然后码了一份复杂而错误的贪心[心情复杂]。做这题之前一定要把这句话先念三遍。

这句话的意思就是，保证任意点之间都有一条路径。而且这是一张有n条边的图，除去从1出发的那条边，不就是一棵树了么？惊不惊喜，意不意外？[开启戏精模式]

然后我们会想到，首都的传送器必须指向自己，否则即使从其它城市走K步到达了首都，从首都走K步后也不可能恰好到达首都。还有一个贪心是，以首都为根节点，要从最深的地方开始向上遍历，修改传送器（想象有这样一棵树，离首都近的地方节点少，越深的地方节点越多，子树也越多，那么显然修改偏上的地方的传送器更优）。

$K>n$和$K=n$是等价的，因为可以在首都绕圈子（首都的传送器指向自己）。

代码还是挺好写的。

代码

#include<iostream>
#include<cstdio>
#define N 100010
using namespace std;
int Ans,K,D[N],Max[N],End[N],Last[N],Next[N],cnt;
void Ins(int x,int y){
    End[++cnt]=y,Next[cnt]=Last[x],Last[x]=cnt;
}
void DFS(int u,int fa){
    D[u]=D[fa]+1,Max[u]=D[u];
    for(int i=Last[u];i;i=Next[i]){
        if(End[i]==fa)continue;
        DFS(End[i],u);
        Max[u]=max(Max[u],Max[End[i]]);
    }
    if(u==1)return;
    if(Max[u]-D[u]>=K-1 && fa!=1)Ans++,Max[u]=0;
}
int main(){
    int n;scanf("%d%d",&n,&K);
    for(int i=1;i<=n;i++){
        int t;scanf("%d",&t);
        if(i==1 && t!=1){Ans++;continue;}
        Ins(t,i);
    }
    DFS(1,0);
    printf("%d",Ans);return 0;
}