随笔

浇 水
  【题目描述】
  瓜农王大爷去年种西瓜赚了不少钱。看到收入不错,今年他又重新开辟了n个西瓜地。
  为了能给他的n个西瓜地顺利的浇上水,对于每个西瓜地他可以选择在本地打井,也可以修管道从另一个瓜地(这个瓜地可能打了井;也可能没打井,他的水也是从其他瓜地引来的)将水引过来。
  当然打井和修管道的费用有差别。已知在第i个西瓜地打井需要耗费wi元,在第i、j个西瓜地之间修管道需要耗费pi,j元。
  现在的问题是:王大爷要想使所有瓜地都被浇上水,至少需要花费多少钱(打井与修管道的费用和)?
  由于瓜地较多,王大爷无法选择在哪些(个)瓜地打井,哪些西瓜地之间修管道。
  请你编程帮王大爷做出决策,求出最小费用。
  【输入格式】
  第1行,一个正整数n,代表西瓜地的数量。
  以下n行,依次给出整数w1..wn(每块西瓜地的打井费用)。
  紧接着是一个n*n的整数矩阵,矩阵的第i行第j列的数代表pi,j(两块西瓜地之间建立管道的费用)。每行的两个数之间有一个空格隔开。
  【输出格式】
  一个正整数,所求的最小花费。
water.in	water.out
6
5
4
4
3
1
20
0 2 2 2 9 9
2 0 3 3 9 9
2 3 0 4 9 9
2 3 4 0 9 9
9 9 9 9 0 9
9 9 9 9 9 0	19

  【样例说明】
  
  在第4个瓜地打井(费用为3),然后将第2,3,4个瓜地与第1个瓜地间修管道(费用分别是2,2,2),这样水可以经过管道从4流向1,然后经1再流向2和3;
  在第5个瓜地打井(费用为1),5和6之间修管道(费用为9)。
  这样一共打了2口井,修了4条管道,能给所有的6个瓜地浇水,费用是:3+2+2+2+1+9=19。 
  
  
  
  【数据规模和约定】
  对于所有数据,1<=N<=300,1<=wi<=100000;pi,i=0,1<=pi,j=pj,i<=100000。
最多需要生成n个最小生成树,因为是稠密图,那么可以采用prim算法求得最小生成树,最小生成树中必然有一口井
,所以对于每个最小生成树需要找出这些最小生成树中的最小距离费用,然后与最小的建井费用相比较,来选择建井还是建造
通道
//#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <iostream>
#include <stack>
#include <queue>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <set>
#include <vector>
#include <cstring>
#include <algorithm>

#define INF 0x3fffffff
#define N 305
#define M (2000010 << 2)
#define LL long long
#define mod 95041567

using namespace std;

struct Node{
    int index;
    int w;
    bool operator < (const Node &p) const{
        return w < p.w;
    }
};

struct Tree{
    int cnt[N];
    int w;
};


Node no[N];
Tree root[N];
int v[N][N], maxx, len_root, n;

void add(int len_node){
    int pos = no[len_node].index;
    for(int i = 0; i < n; ++i)
        root[len_root].cnt[i] = v[pos][i];
    root[len_root].w = no[len_node].w;
    root[len_root].cnt[pos] = 0;
    maxx += no[len_node].w;
    no[len_node].w = 0;
    ++ len_root;
}

void prim()
{
    len_root = 0;
    add(0);
    for(int j = 1; j < n; ++ j){
        int a = INF, b = -1, d = -1;
        int i = 0;
        for(; i < len_root; ++ i){
            int k = 0;
            int c = INF;
            for(int l = 0; l < n; ++ l)
                if(root[i].cnt[l] != 0 && root[i].cnt[l] < c){
                    c = root[i].cnt[l];
                    k = l;
                }
            if(c < a){
                a = c;
                b = k;
                d = i;
            }
        }
        for( i = 0; i < n; ++ i)
            if(no[i].w) break;
        if(no[i].w < a){
            b = no[i].index;
            for(int k = 0; k < len_root; ++ k)
                root[k].cnt[b] = 0;
            add(i);
            for(int k = 0; k < n; ++ k)
                if(! root[0].cnt[k]) root[len_root - 1].cnt[k] = 0;
        }
        else {
            for(int i = 0; i < n; ++ i)
                if(no[i].index == b) {
                    no[i].w = 0;
                    break;
                }
            root[d].w += a;
            maxx += a;
            for(int k = 0; k < n; ++ k)
                root[k].cnt[b] = 0;
            for(int k = 0; k < n; ++ k)
                if(root[d].cnt[k] != 0 && v[b][k] < root[d].cnt[k])
                    root[d].cnt[k] = v[b][k];
        }
    }
}

int main() {
    freopen("in.txt", "r", stdin);
    while(scanf("%d", &n) != EOF){
        for(int i = 0; i < n; ++ i){
            scanf("%d", &no[i].w);
            no[i].index = i;
        }
        sort(no, no + n);
        for(int i = 0; i < n; ++ i)
            for(int j = 0; j < n; ++ j)
                scanf("%d", &v[i][j]);
        maxx = 0;
        prim();
        printf("%d\n", maxx);
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值