版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Leon_liuqinburen/article/details/77936130
[编程题]游历魔法王国
- 热度指数:1217时间限制:1秒空间限制:32768K
- 算法知识视频讲解
魔法王国一共有n个城市,编号为0~n-1号,n个城市之间的道路连接起来恰好构成一棵树。
小易现在在0号城市,每次行动小易会从当前所在的城市走到与其相邻的一个城市,小易最多能行动L次。
如果小易到达过某个城市就视为小易游历过这个城市了,小易现在要制定好的旅游计划使他能游历最多的城市,请你帮他计算一下他最多能游历过多少个城市(注意0号城市已经游历了,游历过的城市不重复计算)。
小易现在在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
思路是贪心
#include<bits/stdc++.h>
#define LL long long
#define MEM(a, b) memset(a, b, sizeof(a))
using namespace std;
const int MAXN = 55;
struct Edge
{
int to;
int next;
};
Edge dataedge[MAXN];
int head[MAXN], dis[MAXN], cnt;
void addEdge(int u, int v)
{
dataedge[cnt].to = v;
dataedge[cnt].next = head[u];
head[u] = cnt++;
}
void dfs(int rt)
{
for(int i = head[rt] ; ~i ; i = dataedge[i].next)
{
dis[dataedge[i].to] = dis[rt] + 1;
dfs(dataedge[i].to);
}
}
int main()
{
int n, l;
scanf("%d %d", &n, &l);
int par;
cnt = 0;
MEM(head, 0xff);
for(int i = 1 ; i < n ; i++)
{
scanf("%d", &par);
addEdge(par, i);
}
dfs(0);
int maxLen = 0;
for(int i = 0 ; i < n ; i++)
maxLen = max(maxLen, dis[i]);
if(l <= maxLen)
printf("%d", l + 1);
else {
int res = n;
res = min(res, maxLen + (l - maxLen) / 2 + 1);
printf("%d\n", res);
}
}