用prim算法实现,T1集合为所有点集合,T2集合表示已经是最小生成树中的点
lowcost[maxn]用来实现集合T1内各顶点到集合T2中各顶点的权值最小边的权值
nearvex[maxn]用来实现集合T1内顶点距集合T2中哪个顶点的距离最小
prim算法的思想:
初始:lowcost[k]=edge[v0][k] , nearvex[k]=v0;其中v0指的是从哪点来构造最小生成树。
当lowcost[i]=-1时我们让它表示i顶点加入到了T2集合,
什么样的情况下我们将i结点加入到T2集合呢?
我们选择那些在T1集合中,距离T2集合中点最小的lowcost[k],将该点k加入到集合T2中,
加入后我们应该改变那些在T1集合中点到T2集合中的最小权值,如果edge[k][i]<lowcost[i]
则将lowcost[i]=edge[k][i];nearvex[i]=k;这就完成了一次操作,进行n-1次后,就能构成一个最小生成树
#include <stdio.h>
#define inf 100000
#define maxn 401
int n;
int lowcost[maxn];
int nearvex[maxn];
int edge[maxn][maxn];
int prim(int u0)
{
int sumweight = 0;
int i, j;
for(i = 1; i <= n; i++) //顶点从1开始
{
lowcost[i] = edge[u0][i];
nearvex[i] = u0;
}
lowcost[u0] = 0;
nearvex[u0] = -1; //lowcost[]为-1时表示该点已经加入T2
for(i = 1; i < n; i++) //将n-1个顶点加入到顶点集合T2
{
int min = inf;
int v = -1;
for(j = 1; j <= n; j++)
{
if(nearvex[j] != -1 && lowcost[j] < min)
{
min = lowcost[j];
v = j;
}
}
if(v != -1)
{
//printf("%d %d %d\n", nearvex[v], v, lowcost[v]);
nearvex[v] = -1; //把v加入T2中
sumweight += lowcost[v];
for(j = 1; j <= n; j++)
{
//更新lowcost[]数组
if(nearvex[j] != -1 && edge[v][j] < lowcost[j])//j在T1集合中&&并且满足v到j的距离小于j到T2中顶点的最小值
{
lowcost[j] = edge[v][j];
nearvex[j] = v;
}
}
}
}
return sumweight;
}
int main()
{
int i, j;
int result;
while(scanf("%d", &n) != EOF)
{
for(i = 1; i <= n; i++)
for(j = 1; j <= n; j++)
{
scanf("%d", &edge[i][j]);
}
result = prim(1);
printf("%d\n", result);
}
return 0;
}