USACO--3.1Agri-Net+prime算法

就是一个prime算法。

代码如下：

/*
ID:15674811
LANG:C++
PROG:agrinet
*/

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

#define maxn 110
#define INF 0x3f3f3f3f

int lowcost[maxn],vis[maxn];
int M[maxn][maxn],n;

int prime()
{
    int sum=0;
    memset(vis,0,sizeof(vis));
    vis[1]=1;
    for(int i=1;i<=n;i++)
        lowcost[i]=M[1][i];
    for(int i=1;i<n;i++)
    {
        int k=0,Min=INF;
        for(int j=1;j<=n;j++)
            if(!vis[j]&&lowcost[j]<Min)
            {
                  Min=lowcost[j];
                  k=j;
            }
        vis[k]=1;
        sum+=Min;
        for(int j=1;j<=n;j++)
            if(!vis[j]&&lowcost[j]>M[k][j])
                lowcost[j]=M[k][j];
    }
    return sum;
}

int main()
{
    freopen("agrinet.in","r",stdin);
    freopen("agrinet.out","w",stdout);
    //freopen("lkl.txt","r",stdin);
    while(scanf("%d",&n)!=EOF)
    {
         for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
               scanf("%d",&M[i][j]);
         int ans=prime();
         printf("%d\n",ans);
    }
  return 0;
}