2558. 从数量最多的堆取走礼物

最新推荐文章于 2024-09-27 11:50:10 发布

longxuan01

最新推荐文章于 2024-09-27 11:50:10 发布

阅读量51

点赞数

文章标签：算法数据结构 c++ leetcode

本文链接：https://blog.csdn.net/longxuan01/article/details/134089890

版权

给你一个整数数组 gifts ，表示各堆礼物的数量。每一秒，你需要执行以下操作：

选择礼物数量最多的那一堆。
如果不止一堆都符合礼物数量最多，从中选择任一堆即可。
选中的那一堆留下平方根数量的礼物（向下取整），取走其他的礼物。
返回在 k 秒后剩下的礼物数量。

示例 1：

输入：gifts = [25,64,9,4,100], k = 4
输出：29
解释：
按下述方式取走礼物：

在第一秒，选中最后一堆，剩下 10 个礼物。
接着第二秒选中第二堆礼物，剩下 8 个礼物。
然后选中第一堆礼物，剩下 5 个礼物。
最后，再次选中最后一堆礼物，剩下 3 个礼物。
最后剩下的礼物数量分别是 [5,8,9,4,3] ，所以，剩下礼物的总数量是 29 。
示例 2：

输入：gifts = [1,1,1,1], k = 4
输出：4
解释：
在本例中，不管选中哪一堆礼物，都必须剩下 1 个礼物。
也就是说，你无法获取任一堆中的礼物。
所以，剩下礼物的总数量是 4 。

提示：

1 <= gifts.length <= 10^3
1 <= gifts[i] <= 10^9
1 <= k <= 10^3

1:暴力法（简单不推荐）

class Solution {
public:
    long long pickGifts(vector<int>& gifts, int k) {
        long long s=0;
        while(k>0){
            k--;
            sort(gifts.begin(),gifts.end());
            gifts[gifts.size()-1]=floor(sqrt(gifts[gifts.size()-1]));
        }
        for(int i=0;i<gifts.size();i++){
            s=s+gifts[i];
        }
        return s;
    }
};

最大堆（难度不高，也很好用）

class Solution {
public:
    long long pickGifts(vector<int>& gifts, int k) {
        priority_queue<int> ab(gifts.begin(), gifts.end());
        long long x,s=0;
        while (k--) {
            x=ab.top(); 
            ab.pop();
            ab.push(floor(sqrt(x)));
        }
        for(int i=0;i<gifts.size();i++){
            s=s+ab.top();
            ab.pop();
        }
        return s;
    }
};

STL的力量（和上面差不多但简易）

class Solution {
public:
    long long pickGifts(vector<int>& gifts, int k) {
    make_heap(gifts.begin(), gifts.end());
            while (k--&& gifts[0]>1) {
            pop_heap(gifts.begin(),gifts.end()); 
            gifts.back()=sqrt(gifts.back());
            push_heap(gifts.begin(), gifts.end());
        }
        return accumulate(gifts.begin(), gifts.end(),0LL);
    }
}；