https://www.luogu.com.cn/problem/P2126
題目背景
mzc與djn的…還沒有衆人皆知,所以我們要來宣傳一下。
題目描述
mzc家很有錢(開玩笑),他家有n個男家丁,現在mzc要將她們全都聚集起來(幹什麼就不知道了)。現在知道mzc與男家丁們互相之間通信的時間,請算出把他們每個人叫到需要的總時間(要重複的哦)。保證能把他們每個人叫到。
輸入格式
第一行有一個數n,表示有n個男家丁。第二行一個數m表示有m條通信路線。之後m行,每行三個數a[i],b[i],c[i],表示第a[i]個男家丁(或mzc)和第b[i]個男家丁(或mzc)通信需要時間(雙向)。ai=0表示mzc。
輸出格式
一行,一個數sum,表示把他們每個人叫到需要的總時間。
輸入輸出樣例
輸入 #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
說明/提示
n<=2300
m<=400000
克魯斯卡爾算法(Kruskal算法)求最小生成樹的三步驟:
1.初始化
for(int i=0; i<=n; ++i) fa[i] = i;//初始化
2.查找
int findFather(int x){//查找
while(x != fa[x])
x = fa[x];
return x;
}
3.合併
void Union(int a, int b) {
int faA = findFather(a);
int faB = findFather(b);
if(faA != faB)
fa [faA] = faB;
}
#include<bits/stdc++.h>
using namespace std;
#define N 400005
int n, m, fa[N], ans;
struct node{
int x, y, w;
}a[N];
bool cmp(node t, node p){
return t.w < p.w;
}
int findFather(int x){//查找
while(x != fa[x])
x = fa[x];
return x;
}
int main(){
cin >> n >> m;
for(int i=1; i<=m; ++i){
cin >> a[i].x >> a[i].y >> a[i].w;
}
sort(a+1, a+m+1, cmp);//按權重從小到大排序
for(int i=0; i<=n; ++i) fa[i] = i;//初始化,先令它們屬於各自不同的集合,父結點都是自己(注意下標從0開始):
for(int i=1; i<=m; ++i){
//合併
int faX = findFather(a[i].x);
int faY = findFather(a[i].y);
if(faX != faY){
fa[faX] = faY;
ans += a[i].w;
}
}
printf("%d", ans);
return 0;
}