LeetCode LCP 30. 魔塔游戏（优先队列）

Michael阿明

发布于 2021-09-06 10:42:02

4770

发布于 2021-09-06 10:42:02

文章被收录于专栏：Michael阿明学习之路

文章目录

- 1. 题目
- 2. 解题

1. 题目

小扣当前位于魔塔游戏第一层，共有 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

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/p0NxJO 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

把怪兽的体力值加入优先队列，并记录其总和
当体力不足时，把体力最大的怪兽移到末尾，移到次数 +1

class Solution {
public:
    int magicTower(vector<int>& nums) {
        long long sum = accumulate(nums.begin(), nums.end(), 1LL);
        // long long 写的 int 报错了好多次。。。 
        if(sum <= 0) return -1; // 体力值不够，不可能过关
        long long v = 1, delta = 0, ans=0;
        priority_queue<int> q;
        for(int i = 0, j; i < nums.size(); ++i)
        {   
            if(nums[i] < 0)
            { // 怪兽，加入优先队列
                q.push(-nums[i]);
                delta = delta-nums[i];//记录所有怪兽的总和（正值）
            }
            else
                v = v+nums[i];//自己的体力
            while(v-delta <= 0 && !q.empty())
            {	// 当自己的体力不够打怪兽的时候，把前面最大的怪兽挪到最后
                int tp = q.top();//怪兽的攻击值
                q.pop();
                delta = delta-tp;//剩余的怪兽的攻击总值
                ans++;//移动一次
            }
        }
        return ans;
    }
};

160 ms 68.8 MB C++