数据结构PTA习题：案例6-1.7 公路村村通 (30分)——最小生成树

案例6-1.7 公路村村通 (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

求最小生成树问题。
由于只需要输出村通的最小成本，我没有给最小生成树创建图，只设了cost记录成本。

C语言实现：

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#define Infinity 10000
struct graph
{
	int Ne;
	int Nv;
	int * * M;
};
typedef struct graph * Graph;
Graph CreateGraph(int N);
int * Visited;//记录顶点是否被收录
int main()
{
	int i;
	int N, M;
	scanf("%d %d", &N, &M);
	Visited = (int *)malloc(N * sizeof(int));
	for (i = 0; i < N; i++)
	{
		Visited[i] = 0;
	}
	int cost = 0;//记录村通需要的成本
	Graph G;
	G = CreateGraph(N);
	G->Ne = M;
	int v1, v2, w;
	for (i = 0; i < M; i++)
	{
		scanf("%d %d %d", &v1, &v2, &w);
		G->M[v1 - 1][v2 - 1] = w;
		G->M[v2 - 1][v1 - 1] = w;
	}
	int * dist;
	dist = (int *)malloc(N * sizeof(int));
	for (i = 0; i < N; i++)//dist数组初始化
	{
		dist[i] = Infinity;
	}
	dist[0] = 0;//从收录顶点1开始
	for (i = 0; i < N; i++)//更新顶点1的邻接点的dist值
	{
		if (G->M[0][i] != 0)
		{
			dist[i] = G->M[0][i];
		}
	}
	int min;
	int index;
	int flag = 0;
	int cnt = 0;//记录已收录的顶点个数
	while (1)
	{
		flag = 0;
		min = Infinity;
		for (i = 0; i < N; i++)//寻找未收录顶点中dist值最小的顶点
		{
			if (dist[i] < min&&Visited[i] == 0)
			{
				flag = 1;
				min = dist[i];
				index = i;
			}
		}
		if (flag == 0) { break; }
		//收录未收录顶点中dist值最小的顶点
		Visited[index] = 1;
		cost = cost + dist[index];
		dist[index] = 0;
		cnt++;
		for (i = 0; i < N; i++)
		{
			if (G->M[index][i] != 0 && Visited[i] == 0)//收录某顶点后，该顶点的邻接点dist值需更新的更新
			{
				if (dist[i] > G->M[index][i])
				{
					dist[i] = G->M[index][i];
				}
			}
		}
	}
	if (cnt == N)//图连通
	{
		printf("%d", cost);
	}
	else//图不连通
	{
		printf("%d", -1);
	}
	return 0;
}
Graph CreateGraph(int N)
{
	int i, j;
	Graph G;
	G = (Graph)malloc(sizeof(struct graph));
	G->Ne = 0;
	G->Nv = N;
	G->M = (int * *)malloc(N * sizeof(int *));
	for (i = 0; i < N; i++)
	{
		G->M[i] = (int *)malloc(N * sizeof(int));
	}
	for (i = 0; i < N; i++)
	{
		for (j = 0; j < N; j++)
		{
			G->M[i][j] = 0;
		}
	}
	return G;
}