实用数据结构---最小生成树(克鲁斯卡尔实现)

时间:2023-01-11 19:04:45
题目描述 Description

学校需要将n台计算机连接起来,不同的2台计算机之间的连接费用可能是不同的。为了节省费用,我们考虑采用间接数据传输结束,就是一台计算机可以间接地通过其他计算机实现和另外一台计算机连接。

为了使得任意两台计算机之间都是连通的(不管是直接还是间接的),需要在若干台计算机之间用网线直接连接,现在想使得总的连接费用最省,让你编程计算这个最小的费用。

输入描述 Input Description

输入第一行为两个整数n,m2<=n<=100000,2<=m<=100000),表示计算机总数,和可以互相建立连接的连接个数。接下来m行,每行三个整数a,b,c 表示在机器a和机器b之间建立连接的话费是c。(题目保证一定存在可行的连通方案, 数据中可能存在权值不一样的重边,但是保证没有自环)

输出描述 Output Description

输出只有一行一个整数,表示最省的总连接费用。

样例输入 Sample Input

3 3

1 2 1

1 3 2

2 3 1

样例输出 Sample Output

2

数据范围及提示 Data Size & Hint

最终答案需要用long long类型来保存

这题就是纯克鲁斯卡尔算法,有些人说是克鲁斯卡尔加上并查集,我认为克鲁斯卡尔的核心就是并查集

先从权值最小的边开始,然后维护整个边集集合,防止出现环

注意代码中的查找带了路径压缩

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 100005
struct Node{
int begin;
int end;
int len;
}a[N];
int pa[N];
bool cmp(Node a, Node b){
return a.len < b.len;
}
int findset(int x){
return pa[x] != x ? pa[x] = findset(pa[x]) : x;
}
int main(){
int m, n;
scanf("%d%d", &n, &m);
int i;
for (i = 1; i <= n; i++)
pa[i] = i;
for (i = 0; i < m; i++)
scanf("%d%d%d", &a[i].begin, &a[i].end, &a[i].len);
sort(a, a + m, cmp);
long long sum = 0;
int ans = 0;
for (i = 0; i < m && ans < n; i++){
int x = findset(a[i].begin);
int y = findset(a[i].end);
if (x != y){
sum += a[i].len;
pa[x] = y;
ans++;
}
}
printf("%lld\n", sum);
return 0;
}