现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。
输入格式:
输入数据包括城镇数目正整数NNN(≤1000\le 1000≤1000)和候选道路数目MMM(≤3N\le 3N≤3N);随后的MMM行对应MMM条道路,每行给出3个正整数,分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见,城镇从1到NNN编号。
输出格式:
输出村村通需要的最低成本。如果输入数据不足以保证畅通,则输出−1-1−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
解析:Prim算法求最小生成树
#include <stdio.h>
#include <stdlib.h>
#define INFINITY 65535
#define MAX 1005
#define ERROR -1
int N, M, A[MAX][MAX], dist[MAX]; //Parent[MAX]; //这题中用不到Parent
int FindMinDist() {
int MinDist = INFINITY, MinV;
for(int V = 0; V < N; V++) {
//如果V未被收录且dist[V]更小
if( dist[V] != 0 && dist[V] < MinDist ) {
MinDist = dist[V];
MinV = V;
}
}
if( MinDist < INFINITY )
return MinV;
else return ERROR;
}
int Prim() {
int TotalWeight, V, W, cnt;
//初始化,默认初始点下标为0
for( V = 0; V < N; V++ ) {
dist[V] = A[0][V];
//Parent[V] = 0; //暂定所有顶点的父结点都是0
}
TotalWeight = 0; //初始化权重和
cnt = 0; //初始化收录的顶点数
dist[0] = 0; //将初始点0收录
cnt++;
//Parent[0] = -1;
while(1) {
//V = 未被收录顶点中dist最小者
V = FindMinDist();
//若V不存在,则结束
if( V == ERROR )
break;
//收录V
TotalWeight += dist[V];
dist[V] = 0;
cnt++;
for( W = 0; W < N; W++)
//若W未被收录且是V的邻接点
if( dist[W] != 0 && A[V][W] < INFINITY )
//如果收录V使得dist[W]变小
if( A[V][W] < dist[W] ) {
dist[W] = A[V][W];
//Parent[W] = V;
}
}
if( cnt < N )
TotalWeight = ERROR;
return TotalWeight;
}
int main(){
int a, b, ans;
scanf("%d %d", &N, &M);
//初始化
for(int i = 0; i < N; i++){
dist[i] = INFINITY;
//Parent[i] = -1;
for(int j = 0; j < N; j++)
A[i][j] = INFINITY;
}
//read
for(int i = 0; i < M; i++) {
scanf("%d %d", &a, &b);
scanf("%d", &A[--a][--b]); //题中编号从1开始
A[b][a] = A[a][b];
}
ans = Prim();
if( ans == ERROR )
printf("-1\n");
else
printf("%d\n", ans);
system("pause");
return 0;
}