codevs 1231 最优布线问题 并查集 解题报告

题目描述 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类型来保存

思路

首先要保证联通，所以我们想到了并查集。这道题有了并查集也就不用再建边跑跑跑了。。。
贪心一下，要最小的权值，所以按照权值从小到大排序一下，然后普通合并并且加上对应的合并的两个点的边权。当合并了n-1条边过后，这个图一定就最小联通了。
总体还是不难。
但是记得开long long！

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<vector>
using namespace std;
const int N=100000+5;
long long ans;
int father[N],k=0,n,m;
struct node 
{
    int u,v;
    long long w;
}ed[N];
bool cmp(node a,node b) 
{
    return a.w<b.w;
}
int getfather(int x) 
{
    if (father[x]!=x) father[x]=getfather(father[x]);
    return father[x];
}
void unionn(int x,int y) 
{
    int r1=getfather(x);
    int r2=getfather(y);
    if (r2!=r1) father[r1]=r2;
}
int main() 
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;i++)
    scanf("%d%d%d",&ed[i].u,&ed[i].v,&ed[i].w);
    for (int i=1;i<=n;i++)
    father[i]=i;
    sort(ed+1,ed+1+m,cmp);
    for (int i=1;i<=m;i++)
    {
        int u=ed[i].u,v=ed[i].v;
        if (getfather(u)!=getfather(v)) 
        {
            unionn(u,v);
            ans+=ed[i].w;
            k++;
        }
        if (k==n-1) break;
    }
    printf("%d\n",ans);
    return 0;
}