最小生成树 ———prim算法 poj 1258

prim算法：所有点组成了一个集合 （笔者称之为全集），先从全集中选出一个点pos放入一个集合（笔者称之为 结果集 ），用一个数组记录pos到剩下点的距离（即 结果集到剩余点的距离），选出最短距离所对应的另一个点 D 加入到结果集中 ，将pos 更新为 D，此时遍历图，若D到剩余点的距离小于low数组中对应的值 ，则更新low数组，再重复选取结果集到剩余点的最短距离 更新low数组，poj 1258  点击打开链接

代码如下：

#include<stdio.h>#include<string.h>int map[120][120],visit[120],low[1100];int n,ans;void prim(){    int pos,i,j;    pos=1;    visit[pos]=1;    for(i=1; i<=n; i++)//第一次对low数组赋值        if(i!=pos)            low[i]=map[pos][i];    for(i=1; i<n; i++)    {        int min=999999;        for(j=1; j<=n; j++)            if(visit[j]==0&&min>low[j])            {                min=low[j];                pos=j;            }            //printf("**%d\n",min);        ans+=min;        visit[pos]=1;        for(j=1; j<=n; j++)//更新low数组；            if(visit[j]==0&&map[pos][j]<low[j])                low[j]=map[pos][j];    }}int main(){    int i,j,tem;    while(scanf("%d",&n)!=EOF)    {        memset(map,9999999,sizeof(map));        memset(visit,0,sizeof(visit));        for(i=1; i<=n; i++)            for(j=1; j<=n; j++)            {                scanf("%d",&tem);                map[i][j]=map[j][i]=tem;            }        ans=0;        prim();        printf("%d\n",ans);    }    return 0;}