CCF CSP 201503-4：网络延时

问题描述

　　给定一个公司的网络，由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

代码如下：

#include <bits/stdc++.h>

using namespace std;

vector<int> v[20008];

void dfs(bool visit[], int &max, int &p, int path, int u){ //path为当前路径长度，u为当前访问节点 
	visit[u] = 1;
	if(max < path){
		max = path;
		p = u;
	}
	for(size_t i = 0; i < v[u].size(); i++){
		int w = v[u][i];
		if(!visit[w]){
			dfs(visit, max, p, path+1, w);
		}
	}
}

//网络延时 
int main(){
	int n, m;
	cin >> n >> m;
	
	for(int i = 2; i <= n; i++){
		int temp;
		cin >> temp;
		v[i].push_back(temp);
		v[temp].push_back(i);
	}
	for(int i = 1; i <= m; i++){
		int temp;
		cin >> temp;
		v[i+10000].push_back(temp);
		v[temp].push_back(i+10000);
	}

    bool v1[20008] = {0}, v2[20008] = {0};
    int max = 0, p = 0;
    dfs(v1, max, p, 0, 1);  //先从节点1开始dfs找到距离最远的点p 
    dfs(v2, max, p, 0, p);  //再从节点p开始进行dfs找到全局最远路径 
    cout << max << endl;
	return 0;		
}