繼續暢通工程
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Problem Description
省政府“暢通工程”的目標是使全省任何兩個村莊間都可以實現公路交通(但不一定有直接的公路相連,只要能間接通過公路可達即可)。現得到城鎮道路統計表,表中列出了任意兩城鎮間修建道路的費用,以及該道路是否已經修通的狀態。現請你編寫程序,計算出全省暢通需要的最低成本。
Input
測試輸入包含若干測試用例。每個測試用例的第1行給出村莊數目N ( 1< N < 100 );隨後的 N(N-1)/2 行對應村莊間道路的成本及修建狀態,每行給4個正整數,分別是兩個村莊的編號(從1編號到N),此兩村莊間道路的成本,以及修建狀態:1表示已建,0表示未建。
當N為0時輸入結束。
Output
每個測試用例的輸出占一行,輸出全省暢通需要的最低成本。
Sample Input
3
1 2 1 0
1 3 2 0
2 3 4 0
3
1 2 1 0
1 3 2 0
2 3 4 1
3
1 2 1 0
1 3 2 1
2 3 4 1
0
Sample Output
3
1
0
題目分析:這個題和hdu 1233差不多,就是加了一個狀態,因此我們可以先判斷每條路的狀態,如果是修好的,就讓它們的成本是0;如果沒修,就讓成本給出的成本。這樣問題的處理就和1233一樣了。
Prim算法:
1、時間復雜度O(n^3):
#include<stdio.h> #include<string.h> #include<algorithm> using namespace std; #define inf 0x3f3f3f3f int map[100][100]; int s[100],vis[100]; int n,m; int prim() { int i,j,t,p,min,cnt,minpos; int ans=0; cnt=0; vis[1]=1; s[cnt++]=1; while(cnt<n) { t=cnt; min=inf; for(i=0;i<t;i++) { p=s[i]; for(j=1;j<=n;j++) { if(!vis[j]&&map[p][j]<min) { min=map[p][j]; minpos=j; } } } ans+=min; s[cnt++]=minpos; vis[minpos]=1; } return ans; } int main() { int i,sum; while(~scanf("%d",&n)&&n) { memset(vis,0,sizeof(vis)); memset(map,inf,sizeof(map)); int b,c,d,sta; m=n*(n-1)/2; for(i=0;i<m;i++) { scanf("%d%d%d%d",&b,&c,&d,&sta); if(sta==0) map[b][c]=map[c][b]=d; else map[b][c]=map[c][b]=0; } sum=prim(); printf("%d\n",sum); } return 0; }
#include<stdio.h> #include<string.h> #include<algorithm> using namespace std; #define inf 0x3f3f3f3f int adj[100]; int vis[100]; int nearest[100]; int map[100][100]; int n,m; int prim() { int i,j,min; int ans=0; vis[1]=1; for(i=2;i<=n;i++) { nearest[i]=map[1][i]; adj[i]=1; } int cnt=n-1; while(cnt--) { min=inf; j=1; for(i=1;i<=n;i++) { if(!vis[i]&&nearest[i]<min) { min=nearest[i]; j=i; } } ans+=map[j][adj[j]]; vis[j]=1; for(i=1;i<=n;i++) { if(!vis[i]&&map[i][j]<nearest[i]) { nearest[i]=map[i][j]; adj[i]=j; } } } return ans; } int main() { int i; while(~scanf("%d",&n)&&n) { int b,c,d,sta; memset(vis,0,sizeof(vis)); memset(map,inf,sizeof(map)); m=n*(n-1)/2; for(i=0;i<m;i++) { scanf("%d%d%d%d",&b,&c,&d,&sta); if(sta==0) map[b][c]=map[c][b]=d; else map[b][c]=map[c][b]=0; } int sum=prim(); printf("%d\n",sum); } return 0; }
#include<stdio.h> #include<string.h> #include<algorithm> using namespace std; struct point { int u; int v; int w; }a[5000]; bool comp(point a1,point a2) { return a1.w<a2.w; } int n,m,father[100],t,cnt; int find(int x) { if(father[x]!=x) father[x]=find(father[x]); return father[x]; } void merge(int x,int y) { int p=find(x); int q=find(y); if(p<q) father[q]=p; else father[p]=q; } void initial() { for(int i=1;i<=n;i++) father[i]=i; } int Kruskal() { int ans=0; sort(a,a+m,comp); for(int i=0;i<m;i++) { int x=find(a[i].u); int y=find(a[i].v); if(x!=y) { ans+=a[i].w; merge(x,y); } } return ans; } int main() { int i,sum,b,c,d,sta; while(~scanf("%d",&n)&&n) { initial(); m=n*(n-1)/2; for(i=0,t=0;i<m;i++) { scanf("%d%d%d%d",&b,&c,&d,&sta); a[i].u=b; a[i].v=c; if(sta==0) a[i].w=d; else a[i].w=0; } sum=Kruskal(); printf("%d\n",sum); } return 0; }