hdu 4514 并查集判断无向图是否有环+树(无环连通图)的直径(两次bfs)

点击打开链接

//树为无环连通图
//并查集判断无向图中是否存在环路+树的直径
//树的直径是指树的最长简单路,使用两次bfs:先任选一个起点bfs找到最长路的终点,再从终点进行bfs,则第二次bfs找到的最长路即为树的直径
#include <iostream>
#include<cstdio>
#include<vector>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
typedef long long LL;
const int maxv=1e5+6;
int parent[maxv];
LL maze[maxv];
int v,e;
LL sum;
int k1;
bool done[maxv];
struct Edge{
int to,weight;
};
vector<Edge>G[maxv];
int find(int x)
{
    return x==parent[x]?x:parent[x]=find(parent[x]);
}
void init()
{
    for(int i=1;i<=v;i++)
        parent[i]=i;
}
void bfs(int k)
{
    sum=0;
    memset(done,false,sizeof(done));
    memset(maze,0,sizeof(maze));
    queue<int>pq;
    while(!pq.empty()) pq.pop();
    pq.push(k);
    k1=k;
    done[k]=true;
    while(!pq.empty())
    {
        int p=pq.front();
        pq.pop();
        for(int i=0;i<G[p].size();i++)
        {
            if(!done[G[p][i].to]) {
                maze[G[p][i].to]=max(G[p][i].weight+maze[p],maze[G[p][i].to]);
                if(sum<maze[G[p][i].to]) {
                        sum=maze[G[p][i].to];
                        k1=G[p][i].to;
                }
                done[G[p][i].to]=true;
                pq.push(G[p][i].to);
            }
        }
    }
}
int main()
{
    while(scanf("%d%d",&v,&e)==2)
    {
        int v1,v2,dist;
        init();
        bool flag=false;
        for(int i=0;i<e;i++)
        {
            scanf("%d%d%d",&v1,&v2,&dist);
            Edge k;
            k.to=v2;
            k.weight=dist;
            G[v1].push_back(k);
            k.to=v1;
            G[v2].push_back(k);
            int p1=find(v1);
            int p2=find(v2);
            if(p1==p2) flag=true;           //存在环路
            else parent[p1]=p2;
        }
        if(flag) printf("YES\n");
        else {
            //求树的直径,连通分量不是一个
            vector<int>field;
            for(int i=1;i<=v;i++)
                 if(parent[i]==i) field.push_back(i);
            LL ans=0;
            for(int i=0;i<field.size();i++)
            {
                bfs(field[i]);
                bfs(k1);
                ans=max(ans,sum);
            }
            cout<<ans<<endl;
            field.clear();
        }
        for(int i=1;i<=v;i++)
            G[i].clear();
    }
    return 0;
}