PTA 公路村村通 C语言 Prim最小生成树

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

输入格式:
输入数据包括城镇数目正整数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

//prim最小生成树
#include
#include
#define MaxSize 1050
int Graph[MaxSize][MaxSize];
int PrimSet[MaxSize];
int N, E;
int S = 0;

void CreateGraph() {//创建无向带权图
	int i = E;
	while (i--) {
		int u, w, v;
		scanf("%d%d%d", &u, &v, &w);
		Graph[u][v] = w;
		Graph[v][u] = w;
	}
}

void Create_iCost(int i, int &min, int &v) { //找到结点i的目前最近结点
	min = 0x3f3f3f;
	v = -1;
	for (int j = 1; j <= N; j++) {
		if (j == i || Graph[i][j] == -1 || PrimSet[j]==1) //若j已加入
			continue;
		if (Graph[i][j] < min) {
			min = Graph[i][j];
			v = j;
		}
	}
}
int Prim(int v) {
	PrimSet[v] = 1; //1~N
	S++;
	int SumCost = 0;
	while (S != N) {
		//找出PrimSet中与其他新点距离最短的那一个
		int min = 0x3f3f3f;
		int temp, p, q;
		int flag = 0;
		for (int i = 1; i <= N; i++) {
			if (PrimSet[i] != 1)
				continue;
			Create_iCost(i, p, q);
			if (p < min) {//若i没有除PrimSet外的连接点，p为0
				min = p;
				temp = q;
				flag = 1;
			}
		}
		if (flag == 0) { //若不连通
			return -1;
		}
		PrimSet[temp] = 1;
		S++;
		SumCost += min;
	}
	return SumCost;
}

int main() {
	scanf("%d%d", &N, &E);
	if (E < N - 1) {//若E
		printf("%d", -1);
		return 0;
	}
	memset(Graph, -1, sizeof(int)*MaxSize * MaxSize);
	CreateGraph();
	int SumCost = Prim(1); //从结点1出发建立最小生成树，Prim算法
	printf("%d", SumCost);
}

第3个点感觉有点极限了(汗。。)