ccf csp
2015.03-04
当信息在电脑、交换机之间传递时,每一步只能通过自己传递到自己所连接的另一台电脑或交换机。请问,电脑与电脑之间传递消息、或者电脑与交换机之间传递消息、或者交换机与交换机之间传递消息最多需要多少步。
第二行包含n - 1个整数,分别表示第2、3、……、n台交换机所连接的比自己上一层的交换机的编号。第i台交换机所连接的上一层的交换机编号一定比自己的编号小。
第三行包含m个整数,分别表示第1、2、……、m台终端电脑所连接的交换机的编号。
1 1 3
2 1
其中电脑1与交换机4之间的消息传递花费的时间最长,为4个单位时间。
1 2 2
3 4 4 4
其中电脑1与电脑4之间的消息传递花费的时间最长,为4个单位时间。
前50%的评测用例满足:n ≤ 20, m ≤ 20。
前70%的评测用例满足:n ≤ 100, m ≤ 100。
所有评测用例都满足:1 ≤ n ≤ 10000,1 ≤ m ≤ 10000。
解题思路:本题考察树的直径的求法,先以任何一个节点以O(n)的复杂度求出该节点到其他节点的距离,找到距离最远的是哪个点,然后再以改点为起始点,以O(n)的复杂度求出改点到其他所有点的距离,此时求出的最大距离即为树的直径。两个过程综合时间复杂度为O(n)。算法的正确性由反证法容易容易证得。
附AC代码如下:
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cstdlib>
#include <queue>
#include <stack>
#include <cmath>
#include <vector>
#define ll long long
#define ull unsigned long long
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
using namespace std;
const int inf = 0x3f3f3f3f;
const int M = 120002;
struct Node{
int v, next;
}node[M*2];
int head[M*2];
int n, m, cnt;
int dist1[M], dist2[M], vis[M];
void addEdge(int u, int v){
node[cnt].v = v;
node[cnt].next = head[u];
head[u] = cnt++;
}
void getDist(int u, int di, int dist[]){
dist[u] = di;
vis[u] = 1;
for(int i = head[u]; i != -1; i = node[i].next){
int v = node[i].v;
if(!vis[v]){
getDist(v, di+1, dist);
}
}
}
int main()
{
// freopen("input.txt", "r", stdin);
while(~scanf("%d%d", &n, &m)){
cnt = 0;
memset(head, -1, sizeof(head));
memset(dist1, 0, sizeof(dist1));
memset(dist2, 0, sizeof(dist2));
memset(vis, 0, sizeof(vis));
for(int i = 2; i <= n; i++){
int u;
scanf("%d", &u);
addEdge(u, i);
addEdge(i, u);
}
for(int i = 1; i <= m; i++){
int u;
scanf("%d", &u);
addEdge(i+n, u);
addEdge(u, i+n);
}
getDist(1, 0, dist1);
int tmp = -1, maxx = -1;
for(int i = 1; i <= n+m; i++){
if(dist1[i] > maxx){
tmp = i;
maxx = dist1[i];
}
}
memset(vis, 0, sizeof(vis));
getDist(tmp, 0, dist2);
maxx = -1;
for(int i = 1; i <= m+n; i++){
if(maxx < dist2[i])
maxx = dist2[i];
}
printf("%d\n", maxx);
}
return 0;
}