暢通工程之最低成本建設問題(最小生成樹(Kruskal)+並查集)
阿新 • • 發佈:2019-01-27
題目連結
某地區經過對城鎮交通狀況的調查,得到現有城鎮間快速道路的統計資料,並提出“暢通工程”的目標:使整個地區任何兩個城鎮間都可以實現快速交通(但不一定有直接的快速道路相連,只要互相間接通過快速路可達即可)。現得到城鎮道路統計表,表中列出了有可能建設成快速路的若干條道路的成本,求暢通工程需要的最低成本。
輸入格式:
輸入的第一行給出城鎮數目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;
}