CCF历届真题练习答案整理（二）

试题编号：    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 ;

}