想象一下(imagine)

【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;
}
原文地址:https://www.cnblogs.com/xjqxjq/p/10544722.html