PTA7-1 公路村村通 分数 20 作者 陈越 单位 浙江大学

现有村落间道路的统计数据表中，列出了有可能建设成标准公路的若干条道路的成本，求使每个村落都有公路连通所需要的最低成本。

输入格式:
输入数据包括城镇数目正整数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
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB

答案：

# include <stdio.h>
# include <stdlib.h>
# define MAXV 1010
# define INF 9999

typedef struct GNode
{
	int V[MAXV];
	int E[MAXV][MAXV];
	int n, e;
}GNode;

GNode* CreateGraph()
{
	GNode* G;
	G = (GNode*)malloc(sizeof(GNode));
	int num1, num2, cost;//输入数据

	scanf("%d %d", &(G->n), &(G->e));
	for (int i = 1; i <= G->n; i++)G->V[i] = i;//顶点集初始化
	for (int i = 1; i <= G->n; i++)//边集初始化
	{
		for (int j = 1; j <= G->n; j++)G->E[i][j] = INF;
	}

	for (int i = 1; i <= G->e; i++)
	{
		scanf("%d %d %d", &num1, &num2, &cost);
		G->E[num1][num2] = cost;
		G->E[num2][num1] = cost;//因为此图为无向图，所以邻接矩阵对称
	}
	return G;
}

void Prim(GNode* G)
{
	int lowcost[MAXV];
	int count = 1;//n个顶点无向图的最小生成树有n-1条边。
	int min, minId, sum = 0;

	for (int i = 1; i <= G->n; i++)lowcost[i] = G->E[1][i];

	lowcost[1] = 0;

	while (count < G->n)
	{
		min = INF;
		for (int i = 1; i <= G->n; i++)
		{
			if (lowcost[i] != 0 && lowcost[i] < min)
			{
				min = lowcost[i];
				minId = i;
			}
		}
		count++;
		lowcost[minId] = 0;
		sum += min;

		for (int i = 1; i <= G->n; i++)
		{
			if (lowcost[i] != 0 && G->E[minId][i] < lowcost[i])lowcost[i] = G->E[minId][i];
		}
	}

	if (sum > INF)printf("-1");
	else printf("%d", sum);
}

int main()
{
	GNode* G;
	G = CreateGraph();
	Prim(G);
	return 0;
}