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[310][310],dis[310],n,s,sumn=0,minx=99999999;
bool vis[310];
int main()
{
    memset(vis,0,sizeof(vis));
    memset(dis,10,sizeof(dis));
    cin>>n;
    for(int i=1;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=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        {
            cin>>a[i][j];
            a[j][i]=a[i][j];
        }
    for(int i=1;i<=n;i++)
    {
        if(a[s][i]<dis[i])
        dis[i]=a[s][i];
    }
    vis[s]=1;
    for(int i=2;i<=n;i++)
    {
        long long minn=999999999,c=0;
        for(int j=1;j<=n;j++)
            if((!vis[j])&&(dis[j]<minn))
            {
                minn=dis[j];
                c=j;
            }
            vis[c]=1;
            //cout<<"!"<<minn<<"!"<<endl;
            sumn+=minn;
            for(int j=1;j<=n;j++)
                if((a[c][j]<dis[j])&&(!vis[j]))
                    dis[j]=a[c][j];
    }
    cout<<sumn<<endl;
    return 0;
}