Prim算法POJ1258

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

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

 #include<stdio.h>
 #include<iostream>
 #include<string.h>
 #include<algorithm>
 using namespace std;
 const int N=;
 #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=;
 memset(vis,,sizeof(vis));
 vis[]=;///从1开始遍历
 pos=;///记录当前结点

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

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

         }
     }
     result+=mi;
     vis[pos]=;
     for(j=;j<=n;j++){///查找pos周围的最短的距离
         if(vis[j]==&&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=;i<=n;i++){
             for(j=;j<=n;j++){
                 scanf("%d",&v);
                 ma[i][j]=ma[j][i]=v;

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