暢通工程(MST,存在得不到MST的情況)
阿新 • • 發佈:2019-02-08
題目描述
省政府“暢通工程”的目標是使全省任何兩個村莊間都可以實現公路交通(但不一定有直接的公路相連,只要能間接通過公路可達即可)。經過調查評估,得到的統計表中列出了有可能建設公路的若干條道路的成本。現請你編寫程式,計算出全省暢通需要的最低成本。輸入描述:
測試輸入包含若干測試用例。每個測試用例的第1行給出評估的道路條數 N、村莊數目M (N, M < =100 );隨後的 N 行對應村莊間道路的成本,每行給出一對正整數,分別是兩個村莊的編號,以及此兩村莊間道路的成本(也是正整數)。為簡單起見,村莊從1到M編號。當N為0時,全部輸入結束,相應的結果不要輸出。
輸出描述:
對每個測試用例,在1行裡輸出全省暢通需要的最低成本。若統計資料不足以保證暢通,則輸出“?”。
輸入
3 3 1 2 1 1 3 2 2 3 4 1 3 2 3 2 0 100
輸出
3 ?注意:該例存在得不到最小生成樹的情況,所以最後我們 對所有節點是否屬於同一個集合進行判斷。
#include<stdio.h> #include<string.h> #include<stdlib.h> #include<math.h> #include<vector> #include<map> #include<set> #include<queue> #include<stack> #include<algorithm> using namespace std; #define INF 0x3f3f3f3f #define N 1000 int Tree[N]; struct edge{ int a,b; int cost ; bool operator< (const edge &A)const{ return cost<A.cost; } }edge[6000]; int findroot(int x)//查詢根節點 { if(Tree[x]==-1) return x; else{ int tmp=findroot(Tree[x]); Tree[x]=tmp; return tmp; } } int main() { int n,m,i,j; while(~scanf("%d%d",&n,&m)&&n!=0) { for( i=1;i<=n;i++) { scanf("%d%d%d",&edge[i].a,&edge[i].b,&edge[i].cost); } sort(edge+1,edge+1+n); for(i=1;i<=m;i++)Tree[i]=-1; int ans=0; int a,b; for( i=1;i<=n;i++){ a=findroot(edge[i].a); b=findroot(edge[i].b); if(a!=b){ Tree[a]=b; ans+=edge[i].cost; } } if(m==1) printf("%d\n",ans ); else { a=findroot(1); for(i=2;i<=m;i++) { b=findroot(i); if(a!=b) {printf("?\n");break;} } if(i==m+1) printf("%d\n",ans ); } } return 0; }