185. [USACO Oct08] 挖水井

185. [USACO Oct08] 挖水井

           输入文件：water.in   输出文件：water.out   简单对比

时间限制：1 s   内存限制：128 MB

农夫约翰决定给他的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）。

请确定农夫约翰为了完成浇灌所有的牧场所需的最小的总花费。

题目名称：water

输入格式：

第1行：一个单独的整数n。

第2..n+1行：第i+1行包含一个单独的整数w_i。

第n+2..2n+1行：第n+1+i行包含n个用空可分开的整数；其中第j个数是p_ij。

输入样例（file water.in）：

4

5

4

4

3

0 2 2 2

2 0 3 3

2 3 0 4

2 3 4 0

 

输入说明：

这里有4个牧场，修井和修管道的代价如图。

输出格式：

第1行：一个单独的整数，表示花费。

输出样例（file water.out）：

9

输出说明：

农夫约翰可以在第4个牧场修井，并且将每个牧场和第一个连接起来，这样，花费是3+2+2+2=9。

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

const int maxn=100001;
int dad[maxn],n,m,u,v,w,b[maxn],tot,q,k,x;

struct Zb{
    int x;int y;int w;
}a[maxn];

int find(int x){//找父亲 
    if(x!=dad[x]) dad[x]=find(dad[x]);
    return dad[x];
}

void unionn(int x,int y){//找祖先，合并为同一祖先 
    int r1=find(x);
    int r2=find(y);
    if(r1!=r2)
    dad[r1]=r2;
}

int cmp(const Zb &a,const Zb &b){//排序 
    if(a.w<b.w) return 1;
    return 0;
}

int main()
{
    freopen("water.in","r",stdin);
    freopen("water.out","w",stdout);
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>b[i];
    }
    for(int i=1;i<=n;i++) dad[i]=i;
    for(int i=1;i<=n;++i)
    {
        for(int j=1;j<=n;++j){
            cin>>x;
            if(x!=0){//将x所代表的连通的点记录下来 
                m++;//记录有几个点是连通的 
                a[m].x=i,a[m].y=j,a[m].w=x;
          }
      }
    }
    for(int i=1;i<=n;i++)//将需要挖井的费用加入 
    {
       m++;
       a[m].x=i;
       a[m].y=n+1;
       a[m].w=b[i];
       m++;
       a[m].x=n+1;
       a[m].y=i;
       a[m].w=b[i];
    }
    sort(a+1,a+m+1,cmp);//将树从小到大排序 
    k=0;
    for(int i=1;i<=m;i++){
        if(find(a[i].x)!=find(a[i].y))
        {
            unionn(a[i].x,a[i].y);
            tot+=a[i].w;//总费用 
            k++;
        }
        if(k==n) break;//当找完所有点的时候，退出 
    }
    cout<<tot<<endl;
    fclose(stdin);fclose(stdout);
    return 0;
}