LCP 30. 魔塔游戏

Every day a Leetcode

题目来源：LCP 30. 魔塔游戏

解法1：反悔贪心

首先，如果 nums 的元素和小于 0，那么即使把所有负数都移到末尾，也无法访问所有房间（最终血量必然小于 1），返回 −1。

否则遍历数组，能加血就尽管加血，要扣血就直接扣血，但如果血量小于 1，我们就「反悔」：从前面的扣血中，拿出一个扣血量最大的数（最小的负数），移到数组的末尾，把之前扣掉的血重新加回来。

注：实际上不需要真的移到末尾，因为我们保证了数组元素和非负，最后血量必然是正数，所以无需再次遍历移到数组末尾的数字。

代码：

class Solution {
public:
    int magicTower(vector<int>& nums) {
        if (accumulate(nums.begin(), nums.end(), 0LL) < 0)
            return -1;

        int ans = 0;
        long long hp = 1;
        // 最小堆
        priority_queue<int, vector<int>, greater<int>> pq;
        for (int &num : nums)
        {
            if (num < 0)
                pq.push(num);
            hp += num;
            if (hp < 1)
            {
                // 反悔
                hp -= pq.top();
                pq.pop();
                ans++;
            }
        }
        return ans;
    }
};

结果：

复杂度分析：

时间复杂度：O(nlogn)，其中 n 为数组 nums 的长度。

空间复杂度：O(n)，其中 n 为数组 nums 的长度。