USACO 2008 Watering hole Prim

题目

题目描述

Farmer John has decided to bring water to his N (1 <= N <= 300) pastures which are conveniently numbered 1..N. He may bring water to a pasture either by building a well in that pasture or connecting the pasture via a pipe to another pasture which already has water.

Digging a well in pasture i costs W_i (1 <= W_i <= 100,000).

Connecting pastures i and j with a pipe costs P_ij (1 <= P_ij <= 100,000; P_ij = P_ji; P_ii=0).

Determine the minimum amount Farmer John will have to pay to water all of his pastures.

POINTS: 400

农民John 决定将水引入到他的n(1<=n<=300)个牧场。他准备通过挖若

干井，并在各块田中修筑水道来连通各块田地以供水。在第i 号田中挖一口井需要花费W_i(1<=W_i<=100,000)元。连接i 号田与j 号田需要P_ij (1 <= P_ij <= 100,000 , P_ji=P_ij)元。

请求出农民John 需要为连通整个牧场的每一块田地所需要的钱数。

输入输出格式

输入格式：

第1 行为一个整数n。

第2 到n+1 行每行一个整数，从上到下分别为W_1 到W_n。

第n+2 到2n+1 行为一个矩阵，表示需要的经费（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 

分析

要将图中的所有点连接起来，求最小的花费，这是一道典型的最小生成树。但是这道最小生成树的最大难点在于本题多了一个“井”的问题。同时挖井也有一定的花费。如何处理这个井就是我们接下来要考虑的问题。

其实我们有一个非常“投机取巧”的方法。我们把Farmer John牌地下天然矿泉水源也考虑作一个点。这个地下水源到其他点路径的权值就是挖井的费用。然后就是一个没有什么特殊操作的裸Prim了！

（最喜欢这种在模板上稍加改动的题目了）

程序

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 300 + 1, INF = 0x3F3F3F3F;
int n, w, p, EdgeCount=  0, Head[MAXN];
struct edge
{
    int Next, Aim, Weight;
}Edge[MAXN*MAXN];
void insert(int u, int v, int w)
{
    Edge[++EdgeCount] = (edge){Head[u], v, w};
    Head[u] = EdgeCount;
}
int Prim()
{
    bool vis[MAXN];
    int MinLen, Count = 1, ans = 0, visited[MAXN];
    visited[Count] = 0;
    vis[0] = 1;
    for (int i = 1; i <= n; i++)
    {
        MinLen = INF;
        for (int j = 1; j <= Count; j++)
            for (int k = Head[visited[j]]; k; k = Edge[k].Next)
                if (!vis[Edge[k].Aim] && Edge[k].Weight < MinLen)
                {
                    MinLen = Edge[k].Weight;
                    p = Edge[k].Aim;
                }
        vis[p] = 1;
        ans += MinLen;
        Count++;
        visited[Count] = p;
    }
    return ans;
}
int main()
{
    //freopen("testdata.in","r",stdin);
    //freopen("data.out","w",stdout);
    memset(Head,0,sizeof(Head));
    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        cin >> w;
        insert(0,i,w);
        insert(i,0,w);
    }
    for (int i = 1; i <= n ; i++)
        for (int j = 1; j <= n; j++)
        {
            cin >> p;
            if (i != j)
                insert(i,j,p);
        }
    cout << Prim() << endl;
    return 0;
}