试题编号: 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
#include
<iostream>
#include
<vector>
#include
<cstring>
#include
<algorithm>
#include
<cstdio>
using
namespace
std
;
const
int
N =
222222
;
int
edgeCount,firstEdge[
N
], to[
N
], length[
N
], nextEdge[
N
];
vector
<
int
> dist;
void
addEdge(
int
u,
int
v,
int
w){
to
[
edgeCount
] = v;
length
[
edgeCount
] = w;
nextEdge
[
edgeCount
] =
firstEdge
[u];
firstEdge
[u] =
edgeCount
++;
}
void
dfs(
int
p,
int
u,
int
d){
dist
[
u
] = d;
for
(
int
iter =
firstEdge
[u]; iter != -
1
; iter =
nextEdge
[iter]){
if
(
to
[iter] != p){
dfs
(u,
to
[iter], d+
length
[iter]);
}
}
}
int
getDiameter(
int
nodeCount,
vector
<
pair
<
pair
<
int
,
int
>,
int
> > edges){
edgeCount
=
0
;
memset
(
firstEdge
, -
1
,
sizeof
(
firstEdge
));
for
(
vector
<
pair
<
pair
<
int
,
int
>,
int
> >::
iterator
iter = edges.
begin
();
iter != edges.
end
();++iter){
addEdge
(iter->
first
.
first
,iter->
first
.
second
, iter->
second
);
addEdge
(iter->
first
.
second
,iter->
first
.
first
, iter->
second
);
}
dist
.
resize
(nodeCount);
dfs
(-
1
,
0
,
0
);
int
u =
max_element
(
dist
.
begin
(),
dist
.
end
()) -
dist
.
begin
();
dfs
(-
1
,u,
0
);
return
*
max_element
(
dist
.
begin
(),
dist
.
end
());
}
int
main(){
int
n, m, v;
vector
<
pair
<
pair
<
int
,
int
>,
int
> > edges;
scanf
(
"%d%d"
,&n,&m);
for
(
int
i=
1
;i<=n-
1
;++i){
scanf
(
"%d"
,&v);
edges.
push_back
(
make_pair
(
make_pair
(v-
1
, i),
1
));
edges.
push_back
(
make_pair
(
make_pair
(i, v-
1
),
1
));
}
for
(
int
i=n;i<=n+m-
1
;++i){
scanf
(
"%d"
,&v);
edges.
push_back
(
make_pair
(
make_pair
(v-
1
, i),
1
));
edges.
push_back
(
make_pair
(
make_pair
(i, v-
1
),
1
));
}
printf
(
"%d\n"
,
getDiameter
(n+m, edges));
return
0
;
}