洛谷 P2126 Mzc家中的男家丁

P2126 Mzc家中的男家丁

题目背景

ｍｚｃ与ｄｊｎ的…还没有众人皆知，所以我们要来宣传一下。

题目描述

ｍｚｃ家很有钱（开玩笑），他家有ｎ个男家丁，现在ｍｚｃ要将她们全都聚集起来（干什么就不知道了）。现在知道ｍｚｃ与男家丁们互相之间通信的时间，请算出把他们每个人叫到需要的总时间（要重复的哦）。保证能把他们每个人叫到。

输入输出格式

输入格式：

 

第一行有一个数n,表示有n个男家丁。第二行一个数m表示有m条通信路线。之后m行,每行三个数a[i],b[i],c[i],表示第a[i]个男家丁(或mzc)和第b[i]个男家丁(或mzc)通信需要时间(双向)。ai=0表示mzc。

 

输出格式：

 

一行，一个数ｓｕｍ，表示把他们每个人叫到需要的总时间。

 

输入输出样例

输入样例#1： 复制

5
12
0 2 15
2 3 20
3 5 13
1 3 29
0 1 30
2 4 21
0 3 23
5 1 48
0 4 17
0 5 27
1 2 43
2 5 41

输出样例#1： 复制

94

说明

ｎ＜＝2300

ｍ＜＝400000

思路：并查集。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXM 2310
#define MAXN 400010
using namespace std;
int n,m,ans;
struct nond{
    int x,y,z;
}edge[MAXN];
int fa[MAXM];
int find(int now){
    if(fa[now]==now)    return fa[now]=now;
    else return fa[now]=find(fa[now]);    
}
int cmp(nond a,nond b){
    return a.z<b.z;
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
        scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].z);
    sort(edge+1,edge+1+m,cmp);
    for(int i=0;i<=n;i++)    fa[i]=i;
    for(int i=1;i<=m;i++){
        int u=find(edge[i].x);
        int v=find(edge[i].y);
        if(u==v)    continue;
        fa[u]=v;
        ans+=edge[i].z;
    }
    cout<<ans;
}