Agri-Net

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

#include<cstdio>
#include<cstring>
#include<algorithm>
#define MAXN 110

const int INF=1<<28;
using namespace std;
int g[MAXN][MAXN];
int dis[MAXN];
bool vis[MAXN];
int n,sum;
bool sprim()
{
    memset(vis,false,sizeof(vis));
    for(int i=1;i<=n;i++)
        dis[i]=INF;
    dis[1]=0;sum=0;
    for(int i=1;i<=n;i++)
    {
        int min=INF,k=0;
        for(int j=1;j<=n;j++)
        {
            if(!vis[j]&&dis[j]<min)
            {
                min=dis[j];
                k=j;
            }
        }
        if(min==INF) return false;
        vis[k]=true;
        sum+=min;
        for(int j=1;j<=n;j++)
        {
            if(!vis[j]&&dis[j]>g[k][j])
            {
                dis[j]=g[k][j];
            }
        }
    }
    printf("%d
",sum);
    return true;
}
int main()
{
    while(scanf("%d",&n)!=EOF){

       for(int i=1;i<=n;i++)
       {
           for(int j=1;j<=n;j++)
           {
               scanf("%d",&g[i][j]);
           }
       }
       sprim();
    }
    return 0;

}