游历魔法王国

N, L = map(int, input().split())

parents = [int(i) for i in input().split()]

nodes_depth = [0] * N

for i in range(N-2):
    nodes_depth[i+1] = nodes_depth[parents[i]] + 1

max_depth = max(nodes_depth)

if L <= max_depth:
    max_cities = L + 1
else:
    max_cities = max_depth + ((L - max_depth) // 2) + 1

if max_cities <= N:
    print(max_cities)
else:
    print(N)
--------------------- 
作者：ChrisMinions 
来源：CSDN 
原文：https://blog.csdn.net/qq_34617032/article/details/78573539 
版权声明：本文为博主原创文章，转载请附上博文链接！

#include <math.h>
using namespace std;

int main(){
    uint cityNum = 0;//题目中的n
    uint maxStep = 0;//题目中的l
    cin>>cityNum;
    cin>>maxStep;
    
    uint parent[10000];
    
    for(int i = 0 ; i < cityNum-1;i++){
        cin>>parent[i];
    }
    uint depth[10000];//用来保存节点深度的数组
    
    for(int j = 0 ; j < cityNum-1;j++){//获取深度为1的节点
        if(parent[j] == 0){
            depth[j] = 1;
        }
    }
    //更新每一座城市所在的节点深度（本题只需要最大节点深度，此处有些多余，循坏比较多，仅供参考）
    for(int k = 1;k<cityNum-1;k++){
        for(int j = 0 ; j < cityNum-1;j++){//获取深度为k的节点
                if(depth[j] == k){
                    for(int i = 0 ; i < cityNum-1;i++){
                        if(parent[i] == j+1){
                            depth[i] = k+1;
                    }
                }
            }
        }
    }
    
    //获取节点深度的最大值
    uint maxDepth = depth[0];//初始化赋值
    for(int i = 0 ; i < cityNum-1;i++){
        if(maxDepth<depth[i]){
            maxDepth = depth[i];
        }
    }
    
    if(maxDepth>=maxStep+1){
        cout<<maxStep+1;
    }else{//考虑回路
        if(cityNum<maxDepth+floor((maxStep-maxDepth)/2)+1){
            cout<<cityNum;//可以游历的城市数量比现有城市数量都大，比较难发生
        }else{
            cout<<(maxDepth+floor((maxStep-maxDepth)/2)+1);
        }
    }
}