POJ1258 Agri-Net【最小生成树】

#include <iostream>
using namespace std;

#define maxn 105
#define inf 0x3f3f3f3f

int map[maxn][maxn],n;

void Prim()
{
    int i,j,d[maxn],vis[maxn],mi,v;
    for(i=1;i<=n;i++)
    {
        d[i]=map[1][i];
        vis[i]=0;
    }
    for(i=1;i<=n;i++)
    {
        mi=inf;
        for(j=1;j<=n;j++)
            if(!vis[j] && d[j]<mi)
            {
                mi=d[j];
                v=j;
            }
        vis[v]=1;
        for(j=1;j<=n;j++)
            if(!vis[j] && d[j]>map[v][j])
                d[j]=map[v][j];
    }
    for(d[0]=0,i=1;i<=n;i++) d[0]+=d[i];
    cout<<d[0]<<endl;
}

int main()
{
    int i,j;
    while(cin>>n)
    {
        for(i=1;i<=n;i++)
            for(j=1;j<=n;j++)
                cin>>map[i][j];
        Prim();
    }
    return 0;
}