题目:
当信息在电脑、交换机之间传递时,每一步只能通过自己传递到自己所连接的另一台电脑或交换机。请问,电脑与电脑之间传递消息、或者电脑与交换机之间传递消息、或者交换机与交换机之间传递消息最多需要多少步。
第二行包含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。
思路:
计算树的直径问题,两边搜索
第一遍找直径的一个端点,第二遍通过这个端点搜索出直径及另一个端点;
简单证明(也可以看看这个证明树的直径(最长路)的详细证明)
首先先说明,在树中任意两点之间的路径有且只有一条 , dis(x,y)表示 x 到 y 的距离;
设想 x 和 y 分别为直径的两个端点,a 为树上任意一点;
假设从 a 搜索找到的最远的点为 b(b是除 x,y,a 后任意一点),
1.若从 a 到 b 的路径上与直径有交点 c,如下图;
由于 a 到 y 只有 1 条路径为 a->c->y, 故 dis(a,b) = dis(a,c)+dis(c,b) > dis(a,c) + dis(c,y) = dis(a,y)
继而有 dis(c,b) > dis(c,y) ==> dis(x,c)+dis(c,b) > dis(x,c) + dis(c,y)
即 dis(x,b) > dis(x,y) , x->y 不是最长的,矛盾,故假设不成立;
2.若从 a 到 b 的路径上与直径无交点,c 为直径上一点,如下图;
易有 dis(a,b) > dis(a,c)+dis(c,y) (距离a的最远点为b)
显然有 dis(a,b)+dis(a,c) > dis(c,y) 即 dis(c,b) > dis(c,y)
所以 dis(x,c)+dis(c,b) > dis(x,c) + dis(c,y)
故 x->y 不是最长路径,矛盾,假设不成立;
综上 b 一定是 x,y 中的一点
代码
#include<iostream>
#include<cstring>
#include<vector>
using namespace std;
#define MAX 21007
vector<int> a[MAX]; //邻接表
int pos,mc; // 远的点和最远距离
bool b[MAX]; //标记是否遍历过
//深度优先搜索整张图,记录下距离begin最远的点和其距离
void dfs(int begin,int c){
//更新最远点
if(c > mc){
mc=c;
pos=begin;
}
//继续往下搜索
for(int i=0;i<a[begin].size();++i){
if(b[a[begin][i]] == false){
b[a[begin][i]]=true;
dfs(a[begin][i],c+1);
}
}
}
int main(){
int n,m;
cin>>n>>m;
for(int i=2,j;i<=n+m;++i){
cin>>j;
a[j].push_back(i);
a[i].push_back(j);
}
//第一遍DFS,找图的直径的一端
mc=0,pos=0;
memset(b,0,sizeof(b));
dfs(n+m,0);
//第二遍BFS,直到直径的一段后,搜索另一端,并记录直径的长度
mc=0;
memset(b,0,sizeof(b));
dfs(pos,0);
cout<<mc<<endl;
return 0;
}