下面讲一个prime算法的实例:
某地区经过对城镇交通状况的调查,得到现有城镇间快速道路的统计数据,并提出“畅通工程”的目标:使整个地区任何两个城镇间都可以实现快速交通(但不一定有直接的快速道路相连,只要互相间接通过快速路可达即可)。现得到城镇道路统计表,表中列出了任意两城镇间修建快速路的费用,以及该道路是否已经修通的状态。现请你编写程序,计算出全地区畅通需要的最低成本。
输入格式:
输入的第一行给出村庄数目N (1);随后的(行对应村庄间道路的成本及修建状态:每行给出4个正整数,分别是两个村庄的编号(从1编号到N),此两村庄间道路的成本,以及修建状态 — 1表示已建,0表示未建。
输出格式:
输出全省畅通需要的最低成本。
输入样例:
4
1 2 1 1
1 3 4 0
1 4 1 1
2 3 3 0
2 4 2 1
3 4 5 0
输出样例:
3
#include<stdio.h> #define INF 9999 int n; bool a[maxn]; int d[maxn],dp[maxn][maxn]; int prime() { int sum=0; for(int i=1; i<=n; i++) { d[i]=dp[1][i]; a[i]=false; } a[1]=true; d[1]=0; for(int i=1; i<n; i++) { int min=INF,k=-1; for(int j=1; j<=n; j++) { if(min>d[j]&&!a[j]) { min=d[j]; k=j; } } a[k]=true; if(k!=-1) { sum+=d[k]; for(int j=1; j<=n; j++) { if(d[j]>dp[k][j]&&!a[j]) { d[j]=dp[k][j]; } } } } return sum; } int main() { scanf("%d",&n); int m=n*(n-1)/2; for(int i=1; i<=n; i++) for(int j=1; j<=n; j++) { dp[i][j]=INF; } for(int i=1; i<=m; i++) { int x,y,z,d; scanf("%d%d%d%d",&x,&y,&z,&d); if(d==1) { dp[x][y]=dp[y][x]=0; } else { dp[x][y]=dp[y][x]=z; } } printf("%d\n",prime()); return 0; }