HDU-1233-還是暢通工程(最小生成樹)
阿新 • • 發佈:2018-11-12
script merge con 暫時 簡單 content ane str input Problem Description
某省調查鄉村交通狀況,得到的統計表中列出了任意兩村莊間的距離。省政府“暢通工程”的目標是使全省任何兩個村莊間都可以實現公路交通(但不一定有直接的公路相連,只要能間接通過公路可達即可),並要求鋪設的公路總長度為最小。請計算最小的公路總長度。
Input
測試輸入包含若幹測試用例。每個測試用例的第1行給出村莊數目N ( < 100 );隨後的N(N-1)/2行對應村莊間的距離,每行給出一對正整數,分別是兩個村莊的編號,以及此兩村莊間的距離。為簡單起見,村莊從1到N編號。
當N為0時,輸入結束,該用例不被處理。 Output 對每個測試用例,在1行裏輸出最小的公路總長度。 Sample Input 3 1 2 1 1 3 2 2 3 4 4 1 2 1 1 3 4 1 4 1 2 3 3 2 4 2 3 4 5 0 Sample Output 3 5
Hint
Huge input, scanf is recommended.
當N為0時,輸入結束,該用例不被處理。 Output 對每個測試用例,在1行裏輸出最小的公路總長度。 Sample Input 3 1 2 1 1 3 2 2 3 4 4 1 2 1 1 3 4 1 4 1 2 3 3 2 4 2 3 4 5 0 Sample Output 3 5
思路:最小生成樹模板題。兩種算法都可以,直接上代碼。
坑點:(C++過,G++超時//暫時不知道為啥)
克魯斯卡爾算法解:
#include<cstdio> #include<algorithm> using namespace std; int bin[205],n; int flag[205]; struct node{ int x,y,val; }a[10005]; int cmp(node x,node y){ if(x.val!=y.val) returnx.val<y.val; } int find(int x) { if(bin[x] == x) return bin[x]; return bin[x] = find(bin[x]); } void merge(int x,int y){ int fx=find(x); int fy=find(y); if(fx!=fy) bin[fx]=fy; } int same(int x,int y){ if(find(x)==find(y)) return 1; return 0; } int main(){while(scanf("%d",&n)&&n){ int i; int m=n*(n-1)/2; for(i=1;i<=n;i++){ bin[i]=i; flag[i]=0; } for(i=0;i<m;i++) scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].val); sort(a,a+m,cmp); int sum=0,cnt=0; for(i=0;i<m;i++){ if(cnt==n-1) break; if(same(a[i].x,a[i].y))continue; sum+=a[i].val; merge(a[i].x,a[i].y); cnt++; } printf("%d\n",sum); } return 0; }
Prim算法解:
#include<cstdio> #include<algorithm> using namespace std; struct node{ int x,y,val; }a[5005]; int cmp(node x,node y){ if(x.val!=y.val) return x.val<y.val; } int main(){ int i,sum,n,m,num; int flag[200];//標記數組表示,目前節點在集合當中。 while(scanf("%d",&n),n) { m=(n*(n-1))/2; for(i=0;i<m;i++) { scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].val); } sort(a,a+m,cmp);//按照len排序 for(i=1;i<=n;i++) flag[i]=0;//初始化標記 flag[a[0].x]=1;//標記第一個節點 sum=0; for(i=0;i<m;i++) { num=flag[a[i].x]+flag[a[i].y]; if(num==1)//如果兩個都沒標記肯定不在這個集合裏,所以不操作,如果兩個都標記了說明兩個都在集合裏了,則不用修路了 { sum+=a[i].val; flag[a[i].x]=flag[a[i].y]=1; i=0;//可能有漏掉的值小的路需要從頭開始找 (當num==0時就會被漏掉) } } printf("%d\n",sum); } return 0; }
HDU-1233-還是暢通工程(最小生成樹)