#267 传送 [贪心]
题解
比赛的时候写了个10分贪心,然后结束前改了一下代码…然后…就彻底0分了……身败名裂QAQ
注意,题目里面有这样一句话:保证从任意城市出发,经过若干次传送,都能到达首都。我读掉了,然后码了一份复杂而错误的贪心[心情复杂]。做这题之前一定要把这句话先念三遍。
这句话的意思就是,保证任意点之间都有一条路径。而且这是一张有n条边的图,除去从1出发的那条边,不就是一棵树了么?惊不惊喜,意不意外?[开启戏精模式]
然后我们会想到,首都的传送器必须指向自己,否则即使从其它城市走K步到达了首都,从首都走K步后也不可能恰好到达首都。还有一个贪心是,以首都为根节点,要从最深的地方开始向上遍历,修改传送器(想象有这样一棵树,离首都近的地方节点少,越深的地方节点越多,子树也越多,那么显然修改偏上的地方的传送器更优)。
和 是等价的,因为可以在首都绕圈子(首都的传送器指向自己)。
代码还是挺好写的。
代码
#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;
}