ccf 201503-4 网络延时 树的直径问题(两次bfs)

点击打开链接

//树为无环连通图
//树的直径是指树的最长简单路,使用两次bfs:先任选一个起点bfs找到最长路的终点,再从终点进行bfs,则第二次bfs找到的最长路即为树的直径
#include <iostream>
#include<cstdio>
#include<vector>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
const int maxv=2*1e4+6;
int v,e,sum,k1;
bool done[maxv];
struct node{
int x,dist;
node(int x,int dist):x(x),dist(dist) {}
};
vector<int>G[maxv];
void bfs(int k)
{
    sum=0;
    memset(done,false,sizeof(done));
    queue<node>pq;
    while(!pq.empty()) pq.pop();
    pq.push(node(k,0));
    k1=k;
    done[k]=true;
    while(!pq.empty())
    {
        node p=pq.front();
        pq.pop();
        sum=max(sum,p.dist);
        for(int i=0;i<G[p.x].size();i++)
        {
            if(!done[G[p.x][i]]) {
                done[G[p.x][i]]=true;
                pq.push(node(G[p.x][i],p.dist+1));
                k1=G[p.x][i];
            }
        }
    }
}
int main()
{
    scanf("%d%d",&v,&e);
    int cnt=2;
    for(int i=0;i<v+e-1;i++)
    {
        int v1;
        scanf("%d",&v1);                //指向父节点
        G[v1].push_back(cnt);
        G[cnt].push_back(v1);
        cnt++;
    }
    bfs(1);             //得到边界节点
    bfs(k1);
    cout<<sum<<endl;
    for(int i=1;i<=v+e;i++)
        G[i].clear();
    return 0;
}