usaco oct09 Watering Hole

Farmer John希望把水源引入他的N (1 <= N <= 300) 个牧场,牧场的编号是1~N.他将水源引入某个牧场的方法有两个,一个是在牧场中打一口井,另一个是将这个牧场与另一个已经有水源的牧场用一根管道相连.
在牧场i中打井的费用是W_i (1 <= W_i <= 100000).
把牧场i和j用一根管道相连的费用是P_ij (1 <= P_ij <= 100000, P_ij = P_ji, P_ii = 0).
请你求出Farmer John最少要花多少钱才能够让他的所有牧场都有水源.

输入：

第1行: 一个正整数N.
* 第2~N+1行: 第i+1行包含一个正整数W_i.
* 第N+2~2N+1行: 第N+1+i行包含N个用空格分隔的正整数,第j个数表示P_ij.

输出：

总共有四个牧场.在1号牧场打一口井需要5的费用,在2或者3号牧场打井需要4的费用,在4号牧场打井需要3的费用.在不同的牧场间建立管道需要2,3或4的费用.

先找到打井花费最少的牧场作为起点，然后将起点到达每个牧场最短距离dis的初值赋为各个牧场打井的花费，最后写一个Prim算法即可.

调试代码我没有删掉..

 #include<iostream>
 #include<cstring>
 #include<cstdio>
 using namespace std;
 long long a[][],dis[],n,s,sumn=,minx=;
 bool vis[];
 int main()
 {
     memset(vis,,sizeof(vis));
     memset(dis,,sizeof(dis));
     cin>>n;
     for(int i=;i<=n;i++)
     {
         cin>>dis[i];
         if(dis[i]<minx)
         {
             minx=dis[i];
             s=i;
             //cout<<"WWW"<<s<<"WWW"<<endl;
         }
     }
     sumn=sumn+dis[s];
     for(int i=;i<=n;i++)
         for(int j=;j<=n;j++)
         {
             cin>>a[i][j];
             a[j][i]=a[i][j];
         }
     for(int i=;i<=n;i++)
     {
         if(a[s][i]<dis[i])
         dis[i]=a[s][i];
     }
     vis[s]=;
     for(int i=;i<=n;i++)
     {
         long long minn=,c=;
         for(int j=;j<=n;j++)
             if((!vis[j])&&(dis[j]<minn))
             {
                 minn=dis[j];
                 c=j;
             }
             vis[c]=;
             //cout<<"!"<<minn<<"!"<<endl;
             sumn+=minn;
             for(int j=;j<=n;j++)
                 if((a[c][j]<dis[j])&&(!vis[j]))
                     dis[j]=a[c][j];
     }
     cout<<sumn<<endl;
     return ;
 }