7-10 公路村村通 (30 分)
现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。
输入格式:
输入数据包括城镇数目正整数N(≤1000)和候选道路数目M(≤3N);随后的M行对应M条道路,每行给出3个正整数,分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见,城镇从1到N编号。
输出格式:
输出村村通需要的最低成本。如果输入数据不足以保证畅通,则输出−1,表示需要建设更多公路。
输入样例:
6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3
输出样例:
12
#include<stdio.h>
#define Maxsize 1000
#define INF 100000
typedef struct {
int edge[Maxsize][Maxsize];
int n,e;
}MGraph;
int Prim(MGraph g){
//flag记录是否为连通图
int min,i,j,u=0,sum=0,flag;
int v=0;
//顶点集合到其他顶点的最小距离
int lowcost[g.n],vset[g.n];
memset(lowcost,0,sizeof(lowcost));
//初始化操作
for(i=0;i<g.n;i++){
lowcost[i]=g.edge[v][i];
vset[i]=0;
}
vset[v]=1;
//算法关键代码
for(i=0;i<g.n-1;i++){
min=INF;flag=0;
for(j=0;j<g.n;j++){
if(vset[j]==0&&lowcost[j]<min){
flag=1;
min=lowcost[j];
u=j;
}
}
//非连通图输出-1
if(flag==0)
return -1;
//继续以u为顶点更新lowcost[]
vset[u]=1;sum+=lowcost[u];
for(j=0;j<g.n;j++){
if(vset[j]==0&&lowcost[j]>g.edge[u][j]){
lowcost[j]=g.edge[u][j];
}
}
}
return sum;
}
int main(){
int a,b,c,i,j,M,N,sum;
MGraph g;
scanf("%d %d",&M,&N);
for(i=0;i<M;i++){
for(j=0;j<M;j++){
if(i==j)
g.edge[i][j]=0;
else
g.edge[i][j]=INF;
}
}
g.n=M;g.e=N;
for(i=0;i<N;i++){
scanf("%d %d %d",&a,&b,&c);
g.edge[a-1][b-1]=c;
g.edge[b-1][a-1]=c;
}
sum=Prim(g);
printf("%d",sum);
}