【CODEVS1231】最优布线问题（最小生成树模板题）

题目链接

1231 最优布线问题

 时间限制: 1 s

 空间限制: 128000 KB

 题目等级 : 白银 Silver

题解

题目描述 Description

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

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

输入描述 Input Description

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

扫描二维码关注公众号，回复： 2798222 查看本文章

输出描述 Output Description

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

样例输入 Sample Input

3 3

1 2 1

1 3 2

2 3 1

样例输出 Sample Output

2

数据范围及提示 Data Size & Hint

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

解题思路：

模板题，注意结果用long long保存就好。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn=1e5+5;
int n,m,pre[maxn],flag=1;
LL ans=0;
struct edge
{
    int x,y;//边的两个顶点
    LL z;//边的权值
}a[maxn];
bool cmp(edge a,edge b)
{
    return a.z<b.z;
}
int find1(int x)
{
    int r=x;
    while(r!=pre[r])
        r=pre[r];
    int i=x,j;
    while(i!=r)
    {
        j=pre[i];
        pre[i]=r;
        i=j;
    }
    return r;
}
void Kruskal()
{
    int f1,f2,k=0;
    for(int i=1;i<=m;i++)
    {
        f1=find1(a[i].x);
        f2=find1(a[i].y);
        if(f1!=f2)
        {
            ans+=a[i].z;
            pre[f2]=f1;//合并不相同的两个集合
            k++;
            if(k==n-1)break;
        }
    }
    if(k<n-1)
    {
        flag=0;
        return;
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        pre[i]=i;
    for(int i=1;i<=m;i++)
        scanf("%d%d%lld",&a[i].x,&a[i].y,&a[i].z);
    sort(a+1,a+m+1,cmp);
    Kruskal();
    if(flag)printf("%lld\n",ans);
    return 0;
}