取水——高级(最小生成树)

题目描述

Famer 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

4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0

样例输出 #1

9

提示

输入数据解释
总共有四个牧场.在1号牧场打一口井需要5的费用,在2或者3号牧场打井需要4的费用,在4号牧场打井需要3的费用.在不同的牧场间建立管道需要2,3或4的费用，Farmer John需要在4号牧场打一口井,然后把所有牧场都用管道连到1号牧场上,总共的花费是3+2+2+2=9.

思路：最小生成树prim算法

与其他最小生成树不同的是，每个节点i可以自己“打井”，打井费用a[i]。

所以枚举每个节点作为开头，并对于该节点i到其他所有节点j都有f[i][j]=min(f[i][j],a[j])

表示到该节点的最小费用为直接打井和造管道的最小值。

然后利用最小生成树的算法即可。

题目链接：取水——高级

代码：

#include<bits/stdc++.h>
using namespace std;
int n,i,j,k,t,sum,min1,min2,a[301],b[301],f[301][301];
bool c[301];
int main()
{
    cin>>n;
    for (i=1;i<=n;i++)
        cin>>a[i];
    for (i=1;i<=n;i++)
        for (j=1;j<=n;j++)
            cin>>f[i][j];
    min2=30000000;
    for (i=1;i<=n;i++)          //枚举以i作为起点
    {
        for (j=1;j<=n;j++)
        {
            b[j]=min(f[i][j],a[j]);        //到j的费用应取打井与打管道中较小的一个
            c[j]=false;
        }
        t=1;
        c[i]=true;
        sum=a[i];
        while (t<n)
        {
            min1=100001;
            for (j=1;j<=n;j++)
                if ((c[j]==false)and(b[j]<min1))
                {
                    min1=b[j];
                    k=j;
                }                             //寻找目前费用最小的一个节点
            c[k]=true;
            t++;
            sum=sum+b[k];
            for (j=1;j<=n;j++)
                b[j]=min(b[j],f[k][j]);        //更新费用
        }
        if (sum<min2)
            min2=sum;
    }
    cout<<min2;
}