【from new_dtoj 3981: 想象一下(imagine)】
问题描述
我们高大的老班举起了有半个他那么高的三角板,说:“你们想象一下——”
于是你就陷入了想象……
有一棵n 个点的树,每个叶子节点上都有一个人,他们按照每秒钟走一条边的速度向树根(节点1)前进。
你可以运用k 次想象之力,让某一个节点(除了根节点)上的所有人瞬间(耗时为0)转移到这个节点的父亲上。
你想知道最少需要多少时间,所有人可以到达根节点。
题解
orzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchmorzchm
这题我们可以发现当一棵树的边数<=k时,可以直接被传递到根节点
所以我们直接把叶子节点往上走,直到边数不超过k,模拟了几次即为ans
具体见代码(辣鸡queue)
#include <cstdio>
const int N=5e5+5;
int s,n,k,fa[N],m,in[N],t,c,q[N];
int main(){
scanf("%d%d",&n,&k);m=n-1;
for (int i=2;i<=n;i++) scanf("%d",&fa[i]),in[fa[i]]++;
for (int i=1;i<=n;i++) if (!in[i]) q[++t]=i;
while(t){
if (k>=m) break;c=0;s++;
for (int i=1;i<=t;i++){
in[fa[q[i]]]--;
if (!in[fa[q[i]]]) q[++c]=fa[q[i]];
}m-=t;t=c;
}
return printf("%d
",s),0;
}