ACM John的农场（最小生成树） C++实现

                                                        John的农场
Description
John是一个农场主，他有几个牧场，为了好好照顾他的牛，他必须在几个牧场之间来回，可糟糕的天气往往使得道路非常泥泞，为此John准备在牧场之间铺一些石子路，这样在下雨天也能快速地从一个牧场到另外一个牧场。但John的资金有限，为了自己能从任一个牧场都通过石子路到达另外一个牧场，他需要好好设计一下线路。请帮助John设计好线路，使得John能从任一个牧场都通过石子路到达另外一个牧场，且线路的费用最低。

输入：

第一行是一个整数K，表示有多少个测试用例，以后每个测试用例占n+1行。每个测试用例的第一行为一个整数n（3<=n<=20），表示有多少个牧场，从第二行开始为一个n*n的矩阵，矩阵元素aij表示从i个牧场到j个牧场的铺路费用。

输出：

每行输出一个测试用例的最小铺路费用。
 
 
Sample Input 
2
6
0 6 1 5 0 0
6 0 5 0 3 0
1 5 0 5 6 4
5 0 5 0 0 2
0 3 6 0 0 6
0 0 4 2 6 0
4
0 1 2 3
1 0 3 4
2 3 0 1
3 4 1 0
 
Sample Output 
15
4

 

#include<iostream>
using namespace std;
#define Infinity 65535
bool flag[21];
int  cost[21][21],lowcost[21],mincost;
int main()
{
 int k;
 cin>>k;
 while(k--)
 {
  int n;
  cin>>n;
  int i,j;
  mincost = 0;
  memset(flag,false,sizeof(flag));
  memset(lowcost,0,sizeof(lowcost));
  for(i=0;i<n;i++)
  {
   for(j=0;j<n;j++)
   {
       cin>>cost[i][j];
    if(cost[0][i]!=0)
    {
      lowcost[i]=cost[0][i];
    }
    else lowcost[i]=Infinity;
   }
  }
  flag[0]=true;
  i=0;
  int key;
  int min;
  while(i<n-1)
  {
   min= Infinity;
   key = 0;
   for(j=1;j<n;j++)
   {
    if(!flag[j]&&lowcost[j]!=0&&min>lowcost[j])
    {
     min = lowcost[j];
     key = j;
    }
   }
   flag[key] = true;
   mincost+=lowcost[key];
   for(j=1;j<n;j++)
   {
    if(!flag[j]&&cost[key][j]!=0&&cost[key][j]<lowcost[j])
     lowcost[j] = cost[key][j];
   }
   i++;
  }
  cout<<mincost<<endl;
 }
 return 0;
}