Kruskal算法优先试用于稀疏图
把n个点的中m条边的边权按照从小到大排序,然后遍历这m条边,如果这一条边两端的两个不连通,就合并,如果连通了,这条边就作废(这里用到了并查集的知识)。一直到用到的边数为n-1条,因为生成一个树只需要n-1条边。
代码:
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
struct edge{
int u;
int v;
int w;
}e[105];
int n, m, pre[105];
int find(int x) //并查集模板
{
int r=x;
while (pre[r]!=r){
r=pre[r];
}
int i=x, j;
while (i!=r){
j=pre[i];
pre[i]=r;
i=j;
}
return r;
}
void join(int x, int y)
{
int fx=find(x), fy=find(y);
if (fx!=fy)
pre[fx]=fy;
}
int cmp(edge a, edge b)
{
return a.w<b.w;
}
int main()
{
int sum=0, count=0;
scanf("%d%d", &n, &m);
for (int i=1; i<=m; i++)
scanf("%d%d%d", &e[i].u, &e[i].v, &e[i].w);
sort(e+1, e+m+1, cmp);//边权由小到大排序
for (int i=1; i<=n; i++)//每个点一开始相互独立
pre[i]=i;
for (int i=1; i<=m; i++){
if (find(e[i].u)!=find(e[i].v)){
join(e[i].u, e[i].v);
count++;
sum+=e[i].w;
}
if (count==n-1)//n-1条边构成树
break;
}
printf("%d", sum);//sum为树的最小边权之和
return 0;
}