比Prim算法更优秀,并且更好理解。
但是需要借助并查集的知识
并查集算法//如果不清楚并查集,可以去看看这篇,顺便a下板子题
Kruskal板子题
算法思路
先把边按照权值进行排序,用贪心的思想优先选取权值较小的边,并依次连接,判断是否这俩个点连通(用并查集来判断是否存在环)然后继续搜。
#include <bits/stdc++.h>
using namespace std;
#define NewNode (ListNode *)malloc(sizeof(ListNode))
#define Mem(a,b) memset(a,b,sizeof(a))
const int N = 2e5 + 50;
const int INF = 0x3f3f3f3f;
const double EPS = 1e-10;
const unsigned long long mod = 998244353;
const int II = 3.1415926535;
typedef long long ll;
typedef unsigned long long ull;
typedef pair <int,int> pii;
struct node
{
ll u,v,val;
}Node[N];//邻接表
bool cmp(node a,node b)
{
return a.val < b.val;
}//结构体排序
ll fa[N],sum,n,m;
int Find(int x)
{
while(x != fa[x])
x = fa[x] = fa[fa[x]];
return x;
}//并查集
void Kruskal()
{
sort(Node,Node+m,cmp);
for(int i = 0;i < m;i++)
{
int eu = Find(Node[i].u),ev = Find(Node[i].v);
if(eu == ev)
continue;
sum += Node[i].val;
fa[ev] = eu;
}
}
int main()
{
std::ios::sync_with_stdio(false);
cin.tie(0),cout.tie(0);
cin >> n >> m;
for(int i = 1;i <= n;i++)
fa[i] = i;
for(int i = 0;i < m;i++)
cin >> Node[i].u >> Node[i].v >> Node[i].val;
Kruskal();
cout << sum << endl;
}