游历魔法王国

Posted 2020-10-11 wsw_seu

魔法王国一共有n个城市,编号为0~n-1号,n个城市之间的道路连接起来恰好构成一棵树。
小易现在在0号城市,每次行动小易会从当前所在的城市走到与其相邻的一个城市,小易最多能行动L次。
如果小易到达过某个城市就视为小易游历过这个城市了,小易现在要制定好的旅游计划使他能游历最多的城市,请你帮他计算一下他最多能游历过多少个城市(注意0号城市已经游历了,游历过的城市不重复计算)。

输入描述:

输入包括两行,第一行包括两个正整数n(2 ≤ n ≤ 50)和L(1 ≤ L ≤ 100),表示城市个数和小易能行动的次数。
第二行包括n-1个整数parent[i](0 ≤ parent[i] ≤ i), 对于每个合法的i(0 ≤ i ≤ n - 2),在(i+1)号城市和parent[i]间有一条道路连接。

输出描述:

输出一个整数,表示小易最多能游历的城市数量。

示例1

输入

5 2
0 1 2 3

输出

3

 1 //根据题意：第一行输入的是城市的个数和能行动的次数
 2 //第二行的值是parent[0]、parent[1]、parent[2]...parent[k]中的值，也就是城市号
 3 //意思是i+1号城市和parent[0]号城市直接有连接。
 4 //例如输入  0  3   1   3   0   5.意思是1（0+1）号城市和0号城市有连接，（1+1）2号城市和3号城市有连接
 5 #include<iostream>
 6 #include<vector>
 7 #include<algorithm>
 8 using namespace std;
 9 
10 int main()
11 {
12     int n,L;
13     vector<int> parent;
14     cin>>n>>L;
15     for(int i=0;i<n-1;i++){
16         int temp;
17         cin>>temp;
18         parent.push_back(temp);
19     }
20     vector<int> dp(n,0);
21     for(int i=0;i<n-1;i++){
22         dp[i+1]=dp[parent[i]]+1;
23     }
24     int maxlen=*max_element(dp.begin(),dp.end());
25     if(L<=maxlen) cout<<L+1;
26     else{
27         //操作数可能非常多，那么最多只能游历n个城市
28         //先找最长的链路，然后减掉最长的，用剩余的操作数去游历短链路。
29         cout<<min(n,1+maxlen+(L-maxlen)/2);
30     }
31 }