usaco Agri-Net

稠密图的最小生成树

使用prim算法

/*
ID:zhdxzwj1
LANG:C++
PROG:agrinet
*/
#include <bits/stdc++.h>
int flag[102][102];
int seen[102]={0};
int parent[102];
int dist[102]={0};
int main()
{
    freopen("agrinet.in","r",stdin);
    freopen("agrinet.out","w",stdout);
    int n;
    scanf("%d",&n);
    int i,j,k;
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
    {
        scanf("%d",&flag[i][j]);
    }

    parent[0]=-1;
    seen[0]=1;
    for(i=0;i<n;i++)
    {
        if(flag[0][i]>=0)
        dist[i]=flag[0][i];
        else
            dist[i]=INT_MAX;
    }i=0;
     int path=0;
    while(1)
    {
        int mi=INT_MAX;
        for(k=0;k<n;k++)
            if(seen[k]==0&&dist[k]<mi)
            {
                mi=dist[k];
                j=k;
            }

        if(j==i)
            break;
        path+=mi;
        seen[j]=1;
        i=j;
        for(k=0;k<n;k++)
        {
            if(seen[k]==0)
            dist[k]=std::min(flag[j][k],dist[k]);
        }
    }
    printf("%d\n",path);
    return 0;
}