用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; }