题目描述
小扣当前位于魔塔游戏第一层,共有 N
个房间,编号为 0 ~ N-1
。每个房间的补血道具/怪物对于血量影响记于数组 nums
,其中正数表示道具补血数值,即血量增加对应数值;负数表示怪物造成伤害值,即血量减少对应数值;0
表示房间对血量无影响。
小扣初始血量为 1,且无上限。假定小扣原计划按房间编号升序访问所有房间补血/打怪,为保证血量始终为正值,小扣需对房间访问顺序进行调整,每次仅能将一个怪物房间(负数的房间)调整至访问顺序末尾。请返回小扣最少需要调整几次,才能顺利访问所有房间。若调整顺序也无法访问完全部房间,请返回 -1。
样例
输入:nums = [100,100,100,-250,-60,-140,-50,-50,100,150]
输出:1
解释:初始血量为 1。至少需要将 nums[3] 调整至访问顺序末尾以满足要求。
````
输入:nums = [-200,-300,400,0]
输出:-1
解释:调整访问顺序也无法完成全部房间的访问。
#### 限制
* `1 <= nums.length <= 10^5`
* `-10^5 <= nums[i] <= 10^5`
----------
### 算法
##### (贪心,堆) $O(n \log n)$
1. 将所有房间的值进行求和,如果小于 0,则无法访问全部房间。
2. 维护当前血量(初始为 0)和一个大根堆。每次令当前血量与房间的值求和,遇到一个负值的房间,将其相反数插入堆中。
3. 如果当前血量小于 0,则从堆中选择一个最大的值(相当于放到末尾),当前血量加上该值,答案加 1。依次循环直到当前血量大于等于 0。
4. 由于之前保证了一定能访问全部房间,所以被放到末尾的房间可以不再考虑。
#### 时间复杂度
* 每个房间最多进堆一次,出堆一次,故总时间复杂度为 $O(n \log n)$。
#### 空间复杂度
* 需要 $O(n)$ 的额外空间维护堆。
#### C++ 代码
define LL long long
class Solution {
public:
int magicTower(vector[HTML_REMOVED]& nums) {
const int n = nums.size();
LL sum = 0;
for (int i = 0; i < n; i++)
sum += nums[i];
if (sum < 0)
return -1;
priority_queue<int> heap;
sum = 0;
int ans = 0;
for (int i = 0; i < n; i++) {
sum += nums[i];
if (nums[i] < 0)
heap.push(-nums[i]);
while (sum < 0 && !heap.empty()) {
sum += heap.top();
heap.pop();
ans++;
}
}
return ans;
}
};
```