Prim算法POJ1258

http://poj.org/problem?id=1258

这道题是最简单的一个啦，，，，

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
const int N=110;
#define M 0x3f3f3f3f///设置无穷大的数M，当两点之间无法联通时设两者之间距离为M
int ma[N][N],low[N],vis[N];///ma记录道路的长度，low记录最短的路径，vis判断是否经过这点
int n;
int prim(){
int i,j,pos,mi,result=0;
memset(vis,0,sizeof(vis));
vis[1]=1;///从1开始遍历
pos=1;///记录当前结点


for(i=1;i<=n;i++)
    if(i!=pos)low[i]=ma[pos][i];///，初始时pos等于1，查找pos周围可以通过的结点中最短的距离并存储在low中


for(i=1;i<n;i++){///遍历n-1次
    mi=M;
    for(j=1;j<=n;j++){
        if(vis[j]==0&&mi>low[j]){
            mi=low[j];
            pos=j;

        }
    }
    result+=mi;
    vis[pos]=1;
    for(j=1;j<=n;j++){///查找pos周围的最短的距离
        if(vis[j]==0&&low[j]>ma[pos][j])
            low[j]=ma[pos][j];
    }
}
return result;
}
int main()
{
    int i,v,j,ans;
    while(~scanf("%d",&n)){
        memset(ma,M,sizeof(ma));
        for(i=1;i<=n;i++){
            for(j=1;j<=n;j++){
                scanf("%d",&v);
                ma[i][j]=ma[j][i]=v;

            }
        }
        ans=prim();
        cout<<ans<<endl;
    }
    return 0;
}