堆:数据备份

题目https://www.acwing.com/problem/content/description/149/
题意:有n+1栋楼,每相邻两两配对搭一条电缆,现资源k条电缆,问选哪k对消耗最小。(一栋楼不难同时被两栋或以上的其他楼相连)
数据范围
2≤n≤100000,
1≤k≤n/2,
0≤s≤1000000000
输入样例:
5 2
1
3
4
6
12
输出样例:
4
思路:这道题目我们首先很容易发现性质,最优解中每两个配对的办公楼一定是相邻的,既然如此的话不妨设置一个数组D,D[i]表示第i个办公楼和第i-1个办公楼之前的距离.那么我们就可以用堆和链表来解决。
代码实现:

#include<iostream>
#include<cstring>
#include<queue>
#include<vector>
#include<set>

using namespace std;
typedef long long ll;
typedef pair<ll, int> PLI;
const int N = 1e5 + 5;

int n, k;
int l[N], r[N];
ll d[N];

//从双链表里删除一个数
void delete_node(int p)
{
    r[l[p]] = r[p];
    l[r[p]] = l[p];
}

int main()
{
    cin >> n >> k;
    for(int i = 0; i < n; i ++ )
        cin >> d[i];
    for(int i = n - 1; ~i; i -- )
        d[i] -= d[i - 1];

    set<PLI> s;
    d[0] = d[n] = 1e15;
    for(int i =0; i <= n; i ++ ){
        l[i] = i - 1;
        r[i] = i + 1;
        if(i >= 1 && i < n)
            s.insert({d[i], i});
    }
    ll res = 0;
    while(k -- ){
        //取出里面的最小值
        set<PLI>::iterator it = s.begin();
        ll v = it -> first;
        int p = it -> second;
        int left = l[p], right = r[p];
        
        //如果这个一对取用了,那么它的左右两点不能再用,必须从set和链表中删除
        s.erase(it);
        s.erase({d[left], left});
        s.erase({d[right], right});
        delete_node(left);
        delete_node(right);

        res += v;
        d[p] = d[left] + d[right] - d[p];
        s.insert({d[p], p});
    }
    cout << res << endl;
    return 0;
}

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值