P2126 Mzc家中的男家丁(克魯斯卡爾算法(Kruskal算法)求最小生成樹)

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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章