acwing kruscal算法,求最小生成树
#include<cstring>#include<iostream>#include<algorithm>using namespace std;//kruskal算法思路//先将每条边按权重进行排序//再从下到大枚举判断边a与边b是否连通,若不连通加入集合内(主要用到并查集的知识判断是否连通)const int N=1e5+10,M=2e5+10;const
·
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
//kruskal算法思路
//先将每条边按权重进行排序
//再从下到大枚举判断边a与边b是否连通,若不连通加入集合内(主要用到并查集的知识判断是否连通)
const int N=1e5+10,M=2e5+10;
const int INF=0x3f3f3f3f;
int n,m; //n个点,m条边
int p[N];
struct Edge{ //开个结构体存储边与权重,并重载<运算符便于之后用sort排序
int a,b,w;
bool operator< (const Edge&W)const //重载运算符的常见写法
{
return w<W.w;
}
}edges[M];
int find(int x) //并查集常见模板
{
if(p[x]!=x) p[x]=find(p[x]);
return p[x];
}
int kruskal()
{
sort(edges,edges+m); //第一步,将边按权重排序
for(int i=1;i<=n;i++) //初始化并查集
{
p[i]=i;
}
int res=0,cnt=0; //res 记录最小生成树边权之和,cnt表示集合内加入了多少个点
for(int i=0;i<m;i++)
{
int a=edges[i].a,b=edges[i].b,w=edges[i].w;
a=find(a),b=find(b); //得出边a边b各自祖宗节点
if(a!=b) //如果a,b不连通,则连通,并更新res,cnt;
{
p[a]=b;
res+=w;
cnt++;
}
}
if(cnt<n-1) return INF;
else return res;
}
int main()
{
cin>>n>>m;
for(int i=0;i<m;i++)
{
int a,b,w;
cin>>a>>b>>w;
edges[i]={a,b,w};
}
int t=kruskal();
if(t==INF) cout<<"impossible"<<endl;
else cout<<t<<endl;
return 0;
}
更多推荐



所有评论(0)