面试OR笔试38——游历城市

1 题目及要求

1.1 题目描述

一共有n个城市，编号为0到n-1号，几个城市之间的道路连接起来恰好构成一棵树。

小毅现在在0号城市，每次行动小毅会从当前所在的城市走到与其相邻的一个城市，且小毅最多能行动L次。

如果小毅到达过某个城市就视为游历过该城市了，小毅现在要制定旅游计划使他能游历最多的城市，请帮他计算以下他最多能游历多少个城市（注意0号城市已经游历了，游历过的城市不重复计算）。

 

 

1）输入：

输入包括两行。

第一行包括两个正整数n ( 2 <= n <= 50 )和L ( 1<= L <= 100 )，表示城市个数和小毅能行动的次数。

第二行包括n-1个整数 city[i] (0<= city[i] < i+1)，对于每个合法的i (0 <= i <= n-2)，在(i+1)号城市和city[i] 之间有一条道路连续。

 

2）输出：

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

 

3）输入范例：

（1）

5 2

0 1 2 3

（2）

6 4

0 1 0 2 1

 

4）输出范例：

（1）

3

 

（2）

4

 

2 解答

2.1 题目分析

既然是树，就有树干，这样其他的分支都可以看作该树干的枝叶。对于树干上的节点，每多走一步就可以多访问一个节点；而对于分支上的节点，每多走两步才可以多访问一个（因为要沿途回去）。

先求出树干的路径长度maxLen。当maxLen不小于L时就沿着该树干走，因此最多访问L+1节点；当maxLen 小于L时，访问主干节点时顺便访问 (L - maxLen) / 2 个枝叶节点（向下取整）,然后把主干上的节点访问完，因此最多访问maxLen + (L - maxLen) / 2 个节点。但最多能访问n个节点。

 

2.2 代码

#include <iostream>
#include <vector>
using namespace std;

// 求最长的路径
int longestPath(const vector<int> &nums) {
	int ml(0), n(nums.size());
	if (n < 1) return ml;
	vector<bool> dt(n, false);
	for (int k1(n - 1),kt;-1 < k1;--k1) {
		if (dt[k1]) continue;					// 标记过的表示有节点经过他, 因此不是最长
		kt = 0;
		for (int k2(k1);-1 < k2 && nums[k2] < k2 + 1;k2 = nums[k2]-1) {
			dt[k2] = true;
			++kt;
		}
		ml = ml < kt ? kt : ml;
	}
	return ml;
}

int numberOfCity(vector<int> &city, int L) {
	int res(1), cn(city.size()), maxLen(longestPath(city));
	res += maxLen < L ? maxLen + (L - maxLen) / 2 : L;
	return res - 2 < cn ? res : cn + 1;
}

int main() {
	int n, l;
	cin >> n >> l;
	if (n < 1) return -1;
	vector<int> city(n - 1);
	for (int k1(0);k1 < n - 1;++k1)
		cin >> city[k1];
	cout << numberOfCity(city,l);
	return 0;
}