题目描述 Description
学校需要将n台计算机连接起来,不同的2台计算机之间的连接费用可能是不同的。为了节省费用,我们考虑采用间接数据传输结束,就是一台计算机可以间接地通过其他计算机实现和另外一台计算机连接。
为了使得任意两台计算机之间都是连通的(不管是直接还是间接的),需要在若干台计算机之间用网线直接连接,现在想使得总的连接费用最省,让你编程计算这个最小的费用。
输入描述 Input Description
输入第一行为两个整数n,m(2<=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类型来保存
思路:模板题
代码如下
//Kruskal算法
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
const int N=100005;
int pre[N];
int n,m;
struct node
{
int u,v,cost;
}edge[N];
void add(int u,int v,int w,int tol)
{
edge[tol].u=u;
edge[tol].v=v;
edge[tol].cost=w;
//tol++;
}
int find(int x)
{
int r=x;
while (pre[r]!=r)
r=pre[r];
int i=x; int j;
while(i!=r)
{
j=pre[i];
pre[i]=r;
i=j;
}
return r;
}
int join(int a,int b)
{
int f1=find(a);
int f2=find(b);
if(f1==f2) return 0;
else
{
pre[f2]=f1;
return 1;
}
}
void init()
{
for(int i=1;i<=n;i++)
pre[i]=i;
}
bool cmp(node a,node b)
{
return a.cost<b.cost;
}
int main()
{
scanf("%d%d",&n,&m);
init();
int x,y,c;
for(int i=0;i<m;i++)
{
scanf("%d%d%d",&x,&y,&c);
add(x,y,c,i);
}
sort(edge,edge+m,cmp);
long long int ans=0;
int flag=0;
for(int i=0;i<m;i++)
{
if(join(edge[i].u,edge[i].v))
{
ans+=edge[i].cost;
flag++;
if(flag==n-1)
break;
}
}
printf("%lld",ans);
}