POJ 1258 Agri-Net 最小生成树 Prim算法

/*
Memory  POJ 1258
Orzmyk
*/
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXN 1000000
using namespace std;
int graph[120][120],n;
int prim()
{
int s=1,pos,d[120],res=0,ans=0,m=1;//m=1,s=1;初始化
bool v[120];
memset(v,false,sizeof(v));//数组初始化
memset(d,0,sizeof(d));
v[s]=true;
for(int i=1;i<=n;i++)
d[i]=MAXN;//pre
while(true)
{
if(m==n)
break;
res=MAXN;
for(int i=2;i<=n;i++)
{
if(!v[i]&&d[i]>graph[s][i])
d[i]=graph[s][i];
if(!v[i]&&res>d[i])
{
res=d[i];
pos=i;
}
}
s=pos;
v[s]=true;
ans+=res;
m++;
}
return ans;
}
int main()
{
while(cin>>n)
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
cin>>graph[i][j];
cout<<prim()<<endl;
}
//while(1);
return 0;
}