leetcode_2558 从数量最多的堆取走礼物

最新推荐文章于 2024-09-27 15:00:48 发布

_不会dp不改名_

最新推荐文章于 2024-09-27 15:00:48 发布

阅读量171

点赞数

分类专栏： # 堆文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/bdn_nbd/article/details/134088597

版权

堆专栏收录该内容

7 篇文章 0 订阅

订阅专栏

文章讲述了如何使用堆数据结构来解决一个编程问题，给定一个数组，每次取最大数并将其开方向下取整后放入堆中，重复k次，最后计算堆中元素之和。提供了两种代码实现，一种是使用额外空间，另一种是在原数组上操作堆元素。

摘要由CSDN通过智能技术生成

1. 题意

给定一个数组，每次从中取走最大的数，返回开根号向下取整送入堆中，最后计算总和。

从数量最多的堆取走礼物

2. 题解

直接用堆模拟即可

2.1 我的代码

用了额外的空间O( n )
priority_queue会自动调用make_heap() 、pop_heap()

class Solution {
public:
    long long pickGifts(vector<int>& gifts, int k) {
        priority_queue<int, vector<int>, less<> > maxHeap(gifts.begin(), gifts.end());

        long long ans = 0;
        for ( int i = 0; i < k; ++i) {
            int p = maxHeap.top();
            if (p == 1) break;
            maxHeap.pop();
            maxHeap.push( (int) sqrt(p));
        }


        while (!maxHeap.empty()) {
            ans += maxHeap.top();
            maxHeap.pop();
        }

        return ans;
    }
};

2.2 更简的代码

直接在原数组上建堆就可以了

class Solution {
public:
    long long pickGifts(vector<int>& gifts, int k) {
        
        make_heap(gifts.begin(), gifts.end(), less<int>());

        for ( int i = 0; i < k; ++i) {
            pop_heap(gifts.begin(), gifts.end() );
            gifts.back() = sqrt(gifts.back());
            push_heap(gifts.begin(), gifts.end());
        }


        return accumulate(gifts.begin(), gifts.end(), 0ll );
    }
};