D - 还是畅通工程 acm 生成树 prim算法

D - 还是畅通工程

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。
Output
对每个测试用例，在1行里输出最小的公路总长度。
Sample
Input
3
1 2 1
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0
Output
3
5
Huge input, scanf is recommended.

思路：
首先，我们要知道构造最小生成树G的Prim算法的基本思想：首先置S={1}，然后。只要S是V的真子集，就做如下的贪心选择：选取满足条件i属于S，j属于V-S，且C[i][j]最小的边，并将顶点j添加到S中，这个过程一直进行到S=V时为止，选取到的所有边恰好构成G的一颗最小生成树。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=101;
int g[maxn][maxn],fix[maxn],last[maxn],key[maxn];
int n,m;
int Prim(int start)
{
	int ret=0;
	memset(key,0x3f,sizeof(key));
	memset(fix,0,sizeof(fix));
	memset(last,0,sizeof(last));
	key[start]=0;//key是代表源节点到这些节点的权值，源节点到源节点为0 
	last[start]=-1;
	for(int i=1;i<=m;i++)//注意这种写法，是n次外层循环，key[start]=0保证第一次是从源节点开始
	{
		int minKey=0x3f3f3f3f,minKeyNode=0;//0x3f3f3f3f代表一个无穷大值
		//1.找到最小代价的顶点
		for(int j=1;j<=m;j++)
		{
			if(!fix[j]&&key[j]<minKey)
			{
				minKey=key[j];
				minKeyNode=j;
			}
		}
		fix[minKeyNode]=true;//代表已经被放入生成树中 
		ret+=key[minKeyNode]; //当前总权值 
		//将找到的最小代价顶点纳入生成树后，当前到其他顶点的权值也进行更新
		for(int j=1;j<=m;j++)
		{
			if(!fix[j]&&g[minKeyNode][j]&&g[minKeyNode][j]<key[j])
			{
				key[j]=g[minKeyNode][j];//更新权值 
				last[j]=minKeyNode; //用来标记该点相邻的顶点 其实这道题不需要这个数组，只需给出总权值
			}
		}

		
	}
	
	return ret;
}
int main()
{
	while(scanf("%d",&m)!=EOF&&m)
	{
		memset(g,0,sizeof(g));
		for(int i=0;i<m*(m-1)/2;i++)
		{
			int x1,x2,wei;
			scanf("%d%d%d",&x1,&x2,&wei);
			g[x1][x2]=wei;
			g[x2][x1]=wei;
		}
		int r=Prim(1);
		
		 printf("%d\n",r);
	}	
} 

时间复杂度分析：n*2n,即O（n^2）.