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;
}