PTA 7-1公路处处通（30分）

题目

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

这个题目很直白，就是最小生成树，u1s1，这个prim算法和dijkstra没什么大区别。在dijkstra中，假设v为出发点，dis[i]表示i到v的最短路径，那么在dijkstra中dis[i]表示i到我们已经确定的部分最小生成树的距离，而且它还可以表示点i是否被收录。这个题目甚至不用叫我们保存解出来的生成树，因此这个题目写起来比较顺利。

#include<stdio.h>
#define MaxSize 1200
#define infinite 100000
#define ERROR 0
int N, M, total_weight = 0;
int graph[MaxSize][MaxSize];
int dis[MaxSize];

int findmin() {
	int index = -1, ans = infinite;
	for (int i = 1;i <= N;i++) {
		if (dis[i] != 0 && dis[i] < ans) {
			ans = dis[i];
			index = i;
		}
	}
	return index;
}
int prime() {
	int cnt = 1; // 对收录点进行计数
	/*默认从1号点开始*/
	dis[1] = 0;
	for (int i = 1;i <= N;i++) {
		if (graph[1][i] != infinite) {
			dis[i] = graph[1][i];
		}
	}
	while (1) {
		int v = findmin();
		if (v == -1) {
			break;
		}
		cnt++;
		total_weight += dis[v];
		dis[v] = 0;
		for (int i = 1;i <= N;i++) {
			if (dis[i] != 0 && graph[v][i] < dis[i]) {
				dis[i] = graph[v][i];
			}
		}
	}
	if (cnt == N) {
		return 1;
	}
	else {
		return ERROR;
	}
}
int main() {
	int left, right, weight;
	int flag;
	scanf("%d%d", &N, &M);
	for (int i = 1;i <= N;i++) {
		dis[i] = infinite;
		for (int j = 1;j <= N;j++) {
			graph[i][j] = infinite;
		}
	}
	for (int i = 1;i <= M;i++) {
		scanf("%d%d%d", &left, &right, &weight);
		graph[left][right] = graph[right][left] = weight;
	}
	flag = prime();
	if (flag) {
		printf("%d", total_weight);
	}
	else {
		printf("-1");
	}
}

运行结果