给定一个公司的网络,由 n 台交换机和 m 台终端电脑组成,交换机与交换机、交换机与电脑之间使用网络连接。
交换机按层级设置,编号为 1 的交换机为根交换机,层级为 1。
其他的交换机都连接到一台比自己上一层的交换机上,其层级为对应交换机的层级加 1。
所有的终端电脑都直接连接到交换机上。
当信息在电脑、交换机之间传递时,每一步只能通过自己传递到自己所连接的另一台电脑或交换机。
请问,电脑与电脑之间传递消息、或者电脑与交换机之间传递消息、或者交换机与交换机之间传递消息最多需要多少步。
输入格式
输入的第一行包含两个整数 n,m,分别表示交换机的台数和终端电脑的台数。
第二行包含 n−1 个整数,分别表示第 2、3、……、n 台交换机所连接的比自己上一层的交换机的编号。第 i 台交换机所连接的上一层的交换机编号一定比自己的编号小。
第三行包含 m 个整数,分别表示第 1、2、……、m 台终端电脑所连接的交换机的编号。
输出格式
输出一个整数,表示消息传递最多需要的步数。
数据范围
前 30% 的评测用例满足:n≤5,m≤5。
前 50% 的评测用例满足:n≤20,m≤20。
前 70% 的评测用例满足:n≤100,m≤100。
所有评测用例都满足:1≤n≤10000,1≤m≤10000。
输入样例1:
4 2
1 1 3
2 1
输出样例1:
4
样例1解释
样例的网络连接模式如下,其中圆圈表示交换机,方框表示电脑:
其中电脑 1 与交换机 4 之间的消息传递花费的时间最长,为 4 个单位时间。
输入样例2:
4 4
1 2 2
3 4 4 4
输出样例2:
4
样例2解释
样例的网络连接模式如下:
其中电脑 1 与电脑 4 之间的消息传递花费的时间最长,为 4 个单位时间。
算法
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 20010, M = N;
int n, m;
int h[N], e[M], ne[M], idx;
int ans;
void add(int a, int b) {
e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}
int dfs(int u) {
int d1 = 0, d2 = 0;
for (int i = h[u]; ~i; i = ne[i]) {
int j = e[i];
int d = dfs(j);
if (d >= d1) d2 = d1, d1 = d;
else if (d >= d2) d2 = d;
}
ans = max(ans, d1 + d2);
return d1 + 1;
}
int main() {
scanf("%d%d", &n, &m);
memset(h, -1, sizeof h);
for (int i = 2; i <= n; i ++ ) {
int a;
scanf("%d", &a);
add(a, i);
}
for (int i = n + 1; i <= n + m; i ++ ) {
int a;
scanf("%d", &a);
add(a, i);
}
dfs(1);
printf("%d\n", ans);
return 0;
}