CCF---201503-4---网络延时---C++----树的直径

试题编号： 201503-4
试题名称： 网络延时
时间限制： 1.0s
内存限制： 256.0MB

问题描述：

问题描述
　　给定一个公司的网络，由n台交换机和m台终端电脑组成，交换机与交换机、交换机与电脑之间使用网络连接。交换机按层级设置，编号为1的交换机为根交换机，层级为1。其他的交换机都连接到一台比自己上一层的交换机上，其层级为对应交换机的层级加1。所有的终端电脑都直接连接到交换机上。
　　当信息在电脑、交换机之间传递时，每一步只能通过自己传递到自己所连接的另一台电脑或交换机。请问，电脑与电脑之间传递消息、或者电脑与交换机之间传递消息、或者交换机与交换机之间传递消息最多需要多少步。
　　
输入格式
　　输入的第一行包含两个整数n, m，分别表示交换机的台数和终端电脑的台数。
　　第二行包含n - 1个整数，分别表示第2、3、……、n台交换机所连接的比自己上一层的交换机的编号。第i台交换机所连接的上一层的交换机编号一定比自己的编号小。
　　第三行包含m个整数，分别表示第1、2、……、m台终端电脑所连接的交换机的编号。
　　
输出格式
　　输出一个整数，表示消息传递最多需要的步数。
　　
样例输入

4 2
1 1 3
2 1

样例输出

4

样例说明
　　样例的网络连接模式如下，其中圆圈表示交换机，方框表示电脑：
　　
其中电脑1与交换机4之间的消息传递花费的时间最长，为4个单位时间。
　　
样例输入

4 4
1 2 2
3 4 4 4

样例输出

4

样例说明
　　样例的网络连接模式如下：
　　
其中电脑1与电脑4之间的消息传递花费的时间最长，为4个单位时间。
　　
评测用例规模与约定
　　前30%的评测用例满足：n ≤ 5, m ≤ 5。
　　前50%的评测用例满足：n ≤ 20, m ≤ 20。
　　前70%的评测用例满足：n ≤ 100, m ≤ 100。
　　所有评测用例都满足：1 ≤ n ≤ 10000，1 ≤ m ≤ 10000。

题目分析

其实就是求树的直接，直接两次bfs搞定。

实现代码

#include<iostream>
#include<cstring>
#include<queue>
using namespace std;

const int SIZE = 20000 + 5;

struct Edge {
	int to, next;
}edges[SIZE<<1];

int head[SIZE], vis[SIZE], dis[SIZE];
int n, m, p, cnt = 0, ans = -1;

void Connect(int from, int to) {
	edges[++cnt].to = to;
	edges[cnt].next = head[from];
	head[from] = cnt;
}

void bfs(int begin) {
	queue<int> q;
	q.push(begin);
	vis[begin] = 1;
	dis[begin] = 0;
	while (q.size()) {
		int now = q.front();
		q.pop();
		for (int i = head[now]; i; i = edges[i].next) {
			int next = edges[i].to;
			if (vis[next]) continue;
			else {
				ans = max(ans, dis[next] = dis[now] + 1);
				q.push(next);
				vis[next] = 1;
			}
		}
	}

}

int main() {
	memset(head, 0, sizeof(head));
	cin >> n >> m;
	for (int i = 2; i <= n + m; i++) {
		cin >> p;
		Connect(p, i);
		Connect(i, p);
	}
	memset(vis, 0, sizeof(vis));
	bfs(1);
	int index = 1;
	for (index = 1; index <= n + m; index++) if (dis[index] == ans) break;
	ans = -1;
	memset(vis, 0, sizeof(vis));
	bfs(index);
	cout << ans << endl;
	return 0;
}