4926-還是暢通工程(最小生成樹) ZCMU
阿新 • • 發佈:2018-12-09
escription
某省調查鄉村交通狀況,得到的統計表中列出了任意兩村莊間的距離。省政府“暢通工程”的目標是使全省任何兩個村莊間都可以實現公路交通(但不一定有直接的公路相連,只要能間接通過公路可達即可),並要求鋪設的公路總長度為最小。請計算最小的公路總長度。
Input
測試輸入包含若干測試用例。每個測試用例的第1行給出村莊數目N ( < 100 );隨後的N(N-1)/2行對應村莊間的距離,每行給出一對正整數,分別是兩個村莊的編號,以及此兩村莊間的距離。為簡單起見,村莊從1到N編號。 當N為0時,輸入結束,該用例不被處理。
Output
對每個測試用例,在1行裡輸出最小的公路總長度。
Sample Input
8
1 2 42
1 3 68
1 4 35
1 5 1
1 6 70
1 7 25
1 8 79
2 3 59
2 4 63
2 5 65
2 6 6
2 7 46
2 8 82
3 4 28
3 5 62
3 6 92
3 7 96
3 8 43
4 5 28
4 6 37
4 7 92
4 8 5
5 6 3
5 7 54
5 8 93
6 7 83
6 8 22
7 8 17
0
Sample Output
82
思路
最小生成樹模板題
程式碼
#include<bits/stdc++.h> #define MAX 10005 using namespace std; struct edge{ int u; int v; int w; }e[MAX]; bool cmp(edge x,edge y) { return x.w<y.w; } int n,m,f[MAX],sum,c; int getf(int v) { if(f[v]==v) return v; else { f[v]=getf(f[v]); return f[v]; } } int Merge(int v,int u) { int t1=getf(v); int t2=getf(u); if(t1!=t2) { f[t2]=t1; return 1; } return 0; } void init() { for(int i=1;i<=n;i++) { f[i]=i; } } int main() { int i; while(~scanf("%d",&n)) { if(n==0) break; if(n==1) { printf("0\n"); continue; } int m=n*(n-1)/2,x; for(i=1;i<=m;i++) { scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w); } sort(e+1,e+m+1,cmp); init(); c=sum=0; for(i=1;i<=m;i++) { if(Merge(e[i].u,e[i].v)) { c+=1; sum+=e[i].w; } if(c==n-1) break; } printf("%d\n",sum); } return 0; }