1. 程式人生 > 實用技巧 >859. Kruskal演算法求最小生成樹

859. Kruskal演算法求最小生成樹

給定一個n個點m條邊的無向圖,圖中可能存在重邊和自環,邊權可能為負數。

求最小生成樹的樹邊權重之和,如果最小生成樹不存在則輸出impossible。

給定一張邊帶權的無向圖G=(V, E),其中V表示圖中點的集合,E表示圖中邊的集合,n=|V|,m=|E|。

由V中的全部n個頂點和E中n-1條邊構成的無向連通子圖被稱為G的一棵生成樹,其中邊的權值之和最小的生成樹被稱為無向圖G的最小生成樹。

輸入格式

第一行包含兩個整數n和m。

接下來m行,每行包含三個整數u,v,w,表示點u和點v之間存在一條權值為w的邊。

輸出格式

共一行,若存在最小生成樹,則輸出一個整數,表示最小生成樹的樹邊權重之和,如果最小生成樹不存在則輸出impossible。

資料範圍

1n1051≤n≤105,
1m21051≤m≤2∗105,
圖中涉及邊的邊權的絕對值均不超過1000。

輸入樣例:

4 5
1 2 1
1 3 2
1 4 3
2 3 2
3 4 4

輸出樣例:

6



思路:①先將所有便從小到大排序,這裡時間複雜度是O(mlogn)
②列舉每一條邊a,b 權重是w
③採用並查集的方法:如果a,b未連通,就將a,b加入到一個集合

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

const int N = 100010, M = 200010
, INF = 0x3f3f3f3f; int n, m; int p[N]; struct Edge { //a,b是節點 w是權重 int a,b,w; bool operator< (const Edge &W)const { return w < W.w; //權重從小到大排序 } }edges[M]; //並查集模板 int find(int x) { if(p[x] != x) p[x] = find(p[x]); return p[x]; } int kruskal() { sort(edges, edges
+ m); //從小到大排序 for(int i = 1; i <= n; i ++) p[i] = i; //初始化並查集 int res = 0, cnt = 0; //res是當前加入到集合中的權重之和,cnt是當前加入的邊數 for(int i = 0; i < m; i ++) //迭代每一條邊 { int a = edges[i].a, b = edges[i].b, w = edges[i].w; a = find(a), b = find(b); //把a b都加入到主通節點 if(a!=b) //如果a b不連通 { p[a] = b; //合併 res += w; //權重相加 cnt++; //對應邊數相加 } } if(cnt < n - 1) return INF; //如果最後邊數<節點數-1 那麼至少有一個點與連通塊無法相連 返回無窮 return res; } int main() { scanf("%d%d", &n, &m); for(int i = 0; i < m; i ++) { int a, b, w; scanf("%d%d%d", &a,&b,&w); edges[i] = {a,b,w}; } int t = kruskal(); if(t == INF) puts("impossible"); else printf("%d\n", t); return 0; }