leetcode 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

思路:
1、计算血量值和,如果是负数,直接返回-1

2、模拟过程,用一个变量 blood 模拟血量,遍历数组,如果遇到负值就放入优先队列(大根堆),不断更新血量,如果blood < 0, 也就是快死之前,取出堆顶的元素(扣最多的血)给自己加上,表示将这个元素移动到末尾,这样的贪心思想能保证我们移动到尾部的元素是最少的。

class Solution {
public:
    int magicTower(vector<int>& nums) {
        //注意:accumulate默认返回int
        // long long sum = accumulate(nums.begin(), nums.end(), 1);
        long long sum = 1;
        for(int& x : nums){
            sum += x;
        }
        if(sum <= 0) return -1;
        priority_queue<int> p;
        long long blood = 1;
        long long cnt = 0;
        for(int& num : nums){
            //这里将扣掉的血量值放进队列,正值
            if(num < 0) p.push(-num);
            blood += num;
            while(blood <= 0){
                int cur = p.top();
                p.pop();
                //将之前消耗的血量加上,所以之前那个‘消耗较多血量’的位置需要置末尾
                blood += cur;   
                cnt++;  //次数++
            }
        }
        return cnt;
    }
};

你可能感兴趣的:(Leetcode)