poj 1258 Agri-Net

#include<iostream>            //最小生成树prim算法
using namespace std;
const int MAXN=105;    //最大顶点数
int n,edge[MAXN][MAXN];        //n记录顶点数
struct MST    //最小生成树的边
{
    int s,t,w;
}mst[MAXN];
void Prim()
{
    int sum=0,i,j,k;    //sum记录最小生成树的路径和
    for(i=0;i<n-1;i++)    //顶点下标从0开始,默认选择顶点0加入生成树
    {
        mst[i].s=0;mst[i].t=i+1;
        mst[i].w=edge[0][i+1];
    }
    for(i=0;i<n-1;++i)        //求n-1条边
    {
        int min_w=INT_MAX;
        for(j=i;j<n-1;++j)
        {
            if(mst[j].w<min_w)        //求最小权值边
            {
                min_w=mst[j].w;
                k=j;
            }
        }    
        swap(mst[k],mst[i]);    //最小边移至前端
        int v=mst[i].t;
        sum+=mst[i].w;    
        for(j=i+1;j<n-1;++j)    //更新后面的边
        {    
            int wei=edge[v][mst[j].t];
            if(wei<mst[j].w)
            {
                mst[j].w=wei;
                mst[j].s=v;
            }
        }
    }
    cout<<sum<<endl;
}
int main()
{
    while(cin>>n)
    {
        for(int i=0;i<n;++i)
            for(int j=0;j<n;++j)
                cin>>edge[i][j];
        Prim();
    }
    return 0;
}