【反悔堆】力扣LCP 30. 魔塔游戏

加粗样式小扣当前位于魔塔游戏第一层,共有 N 个房间,编号为 0 ~ N-1。每个房间的补血道具/怪物对于血量影响记于数组 nums,其中正数表示道具补血数值,即血量增加对应数值;负数表示怪物造成伤害值,即血量减少对应数值;0 表示房间对血量无影响。

小扣初始血量为 1,且无上限。假定小扣原计划按房间编号升序访问所有房间补血/打怪,为保证血量始终为正值,小扣需对房间访问顺序进行调整,每次仅能将一个怪物房间(负数的房间)调整至访问顺序末尾。请返回小扣最少需要调整几次,才能顺利访问所有房间。若调整顺序也无法访问完全部房间,请返回 -1。

示例 1:
输入:nums = [100,100,100,-250,-60,-140,-50,-50,100,150]

输出:1

解释:初始血量为 1。至少需要将 nums[3] 调整至访问顺序末尾以满足要求。

示例 2:
输入:nums = [-200,-300,400,0]

输出:-1

解释:调整访问顺序也无法完成全部房间的访问。

提示:
1 <= nums.length <= 10^5
-10^5 <= nums[i] <= 10^5

反悔堆

class Solution {
public:
    int magicTower(vector<int>& nums) {
        priority_queue<int, vector<int>, greater<int>> q;
        int ans = 0;
        long long hp = 1, delay = 0;
        for(int num : nums){
            if(num < 0) q.push(num);
            hp += num;
            if(hp <= 0){
                ans++;
                int a = q.top();
                delay += a;
                q.pop();
                hp -= a;
            }
        }
        hp += delay;
        return hp > 0 ? ans : -1;
    }
};

时间复杂度:O(nlogn)。数组 nums 中的每个元素至多进入和退出优先队列各一次,每一次优先队列操作的时间复杂度为 O(logn)。

空间复杂度:O(n),即为优先队列需要的空间。

可以使用最小堆来储存怪物的伤害,由于怪物的伤害num是负数,所以最小堆的堆顶就是怪物造成的最多的伤害。

我们升序遍历每一个房间,并且记录当前的血量。如果当前房间是怪物房间,则将它放到堆中,我们如果受到当前房间效果后生命值不为正,那么就说明我们要将之前的最大怪物伤害的房间放到末尾,所以我们就将最小堆堆顶的值加到hp上,并且ans++代表我们进行了一次调整,然后用delay来储存我们调整到末尾的伤害是多少。当遍历完成后,如果hp += delay不为正数,那么返回-1,否则返回ans。

你可能感兴趣的:(精选,贪心算法,数据结构,leetcode,游戏,算法)