1. 程式人生 > >暢通工程之最低成本建設問題(最小生成樹(Kruskal)+並查集)

暢通工程之最低成本建設問題(最小生成樹(Kruskal)+並查集)

題目連結
某地區經過對城鎮交通狀況的調查,得到現有城鎮間快速道路的統計資料,並提出“暢通工程”的目標:使整個地區任何兩個城鎮間都可以實現快速交通(但不一定有直接的快速道路相連,只要互相間接通過快速路可達即可)。現得到城鎮道路統計表,表中列出了有可能建設成快速路的若干條道路的成本,求暢通工程需要的最低成本。

輸入格式:

輸入的第一行給出城鎮數目N (1<N≤1000)和候選道路數目M≤3N;隨後的M行,每行給出3個正整數,分別是該條道路直接連通的兩個城鎮的編號(從1編號到N)以及該道路改建的預算成本。

輸出格式:

輸出暢通工程需要的最低成本。如果輸入資料不足以保證暢通,則輸出“Impossible”。

輸入樣例1:

6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3

輸出樣例1:

12

輸入樣例2:

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

輸出樣例2:

Impossible

#include<algorithm>
#include<iostream>
#include<vector>
#include<cstdio>
using namespace
std; int f[1001]; typedef struct E{ int xx; int yy; int zz; }E; E edge[3001]; bool cmp(E a,E b){ return a.zz<b.zz; } int getf(int x){ if(f[x] == x) return x; f[x] = getf(f[x]); return f[x]; } int merge(int x,int y){ int tx = getf(x); int ty = getf(y); if
(tx!=ty){ f[ty] = tx; return 1; } return 0; } int main(){ int n,e; scanf("%d%d",&n,&e); for(int i = 1;i<=n;i++) f[i] = i; for(int i = 0;i<e;i++) scanf("%d%d%d",&edge[i].xx,&edge[i].yy,&edge[i].zz); //排序 sort(edge,edge+e,cmp); int cnt = 0;//計算邊數 int sum = 0; for(int i = 0;i<e;i++){ if(merge(edge[i].xx,edge[i].yy)){ cnt++; sum += edge[i].zz; } if(cnt==n-1)//選擇了n-1條邊 break; } int sm = 0; for(int i = 1;i<=n;i++) f[i] == i?sm++:sm; if(sm>1||!sm) printf("Impossible\n"); else printf("%d\n",sum); return 0; }