算法思路
- 将图中的所有边都去掉
- 将边按权值从小到大的顺序添加到图中,保证添加的过程中不会形成环
- 重复上一步直到连接所有顶点,此时就生成了最小生成树
应用了贪心思想
代码
代码中用了并查集,可以查看并查集
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
struct edge //边
{
int from,to,dis;
}e[200001];
int fa[100001]; //父节点
int n,m,cnt,ans;
bool cmp(edge a,edge b)
{
return a.dis<b.dis; //用于排序
}
int getfa(int i) //寻找根节点
{
if(fa[i]==i)
return i;
return getfa(fa[i]);
}
void unionn(int u,int v)
{
fa[getfa(u)]=getfa(v); //合并
}
void kruskal()
{
sort(e+1,e+1+m,cmp); //排序
for(int i=1;i<=m;i++)
{
if(cnt==n-1) //共n-1条边
return;
if(getfa(e[i].from)!=getfa(e[i].to))//如果根节点相同,则说明已经成环
{
unionn(e[i].to,e[i].from); //使用一条边后,将两端点设置为子节点和父节点
ans+=e[i].dis;
cnt++;
}
}
}
int main()
{
cin>>n>>m;
for(int i=1;i<=m;i++)
scanf("%d%d%d",&e[i].from,&e[i].to,&e[i].dis);
kruskal();
cout<<ans;
return 0;
}