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).