CCF CSP 2015年03月第4题 网络延迟（求树的直径）

问题描述

试题编号：	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 样例说明 　　样例的网络连接模式如下，其中圆圈表示交换机，方框表示电脑： 　　其中电脑1与交换机4之间的消息传递花费的时间最长，为4个单位时间。 样例输入 4 4 1 2 2 3 4 4 4 样例输出 4 样例说明 　　样例的网络连接模式如下： 　　其中电脑1与电脑4之间的消息传递花费的时间最长，为4个单位时间。 评测用例规模与约定 　　前30%的评测用例满足： n  ≤ 5,   m  ≤ 5。 　　前50%的评测用例满足： n  ≤ 20,   m  ≤ 20。 　　前70%的评测用例满足： n  ≤ 100,   m  ≤ 100。 　　所有评测用例都满足：1 ≤   n  ≤ 10000，1 ≤   m  ≤ 10000。

解题思路：求一棵树上两个点之间的最常距离，使用两次BFS即可。第一次BFS从任意一点开始，找到距离该点最远的一个点S；然后再从S出发，进行第二次BFS，找到距离S点最远的点T。S和T之间的距离就是最长的。

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <vector>

#define MAXN 20005

using namespace std;
vector<int> G[MAXN];
queue<int> que;
int dis[MAXN];
bool vis[MAXN];
int n,m;

void init() {
    for(int i = 0; i < n+m; i++) {
        G[i].clear();
    }
}

void add_edge(int from, int to) {
    G[from].push_back(to);
    G[to].push_back(from);
}

int bfs(int &s) {
    while(que.size())que.pop();
    memset(dis,0,sizeof(dis));
    memset(vis,0,sizeof(vis));
    que.push(s);
    vis[s] = true;
    int ans = 0;
    while(que.size()) {
        int u = que.front();
        que.pop();
        for(int i = 0; i < (int)G[u].size(); i++) {
            int v = G[u][i];
            if(!vis[v]) {
                que.push(v);
                vis[v] = true;
                dis[v] = dis[u] + 1;
                if(dis[v] > ans) {
                    ans = dis[v];
                    s = v;
                }
            }
        }
    }
    return ans;
}

int main(void) {
    while(scanf("%d %d",&n,&m) != EOF) {
        int x;
        init();
        for(int i = 2; i <=n; i++) {
            scanf("%d",&x);
            add_edge(i,x);
        }
        for(int i = 1; i <= m; i++) {
            scanf("%d",&x);
            add_edge(x,i+n);
        }
        int s = 1;
        int ans = max(bfs(s),bfs(s));
        printf("%d\n",ans);
    }
    return 0;
}