网易笔试--游历魔法王国

好久没写博客了，拿了一道网易的笔试题做，书归正传：

 

题目

时间限制：1秒
空间限制：32768K
魔法王国一共有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

输出例子1:
3

这道题乍一看有点像并查集的问题（扫一眼确实挺像的、、），分析之后其实就是给出一棵树，让你求这棵树的深度是多少，而且这棵树的构造还是倒着的、、（孩子的指针指向父亲），所以就写了代码，递归实现查看一棵树的深度，如果深度大于等于m+1结果就是m+1，否则就是这棵树的深度。。。那么问题来了，输入

3 3

0 0

结果应该是3、、也就是说这道题在主要这埋下了伏笔，就是走到头了还能在走回来、、

所以由于题目给出的输入数据个数很少，时间又太晚了、、就泛泛的给出代码（数据量大一点跑起来肯定吃力，可以采用更合适的数据结构来重新构思这个问题）

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
using namespace std;

const int N = 120;
int depth[N] , father[N];
bool vis[N];
int res = 1;
int res_index = 0;
void init(int n){
	memset(depth , 1 , sizeof(depth));
	memset(father, 0 , sizeof(father));
	memset(vis,false,sizeof(vis));
	depth[0] = 1;
	vis[0] = true;
	for(int i = 1;  i< n; ++i){
		cin>>father[i];
	}
}

int update(int index){
	if(vis[index]){
		return depth[index];
	}
	int fa = father[index];
	int dep = update(fa);
	depth[index] = dep + 1;

	if(res < depth[index]){
		res = depth[index];
		res_index = index;
	}
	vis[index] = true;
	return depth[index];
}

int main(){
	int n , m;
	while(cin>>n>>m){
		init(n);
		for(int i = 1 ; i<n; ++i){
			update(i);
		}
		if(m + 1 <= res){
			cout<<m + 1<<endl;
		}else{
			//这里可以折回来再走
			m = m - res + 1;	//还可以走多少步
			int index = res_index;
			memset(vis,false,sizeof(vis));
			vis[0] = true;
			while(m>0){
				--m;
				bool flag=false;
				for(int i = 0 ;i < n; ++i)if(father[i]==index && !vis[i]){
					flag = true;
					vis[index] = true;
					index = i;
					++res;
					break;
				}
				if(!flag){
					index = father[index];
					vis[index] = true;
				}
				
			}cout<<res<<endl;
			 
		}
	}
	return 0 ;
}

　　

转载于:https://www.cnblogs.com/dqsBK/p/7503014.html