小扣当前位于魔塔游戏第一层,共有
N
个房间,编号为0 ~ N-1
。每个房间的补血道具/怪物对于血量影响记于数组nums
,其中正数表示道具补血数值,即血量增加对应数值;负数表示怪物造成伤害值,即血量减少对应数值;0
表示房间对血量无影响。小扣初始血量为 1,且无上限。假定小扣原计划按房间编号升序访问所有房间补血/打怪,为保证血量始终为正值,小扣需对房间访问顺序进行调整,每次仅能将一个怪物房间(负数的房间)调整至访问顺序末尾。请返回小扣最少需要调整几次,才能顺利访问所有房间。若调整顺序也无法访问完全部房间,请返回 -1。
class Solution {
public:
int magicTower(vector& nums) {
}
};
LCP 30. 魔塔游戏
对于数组和小于等于0的情况直接返回-1
然后我们贪心的考虑,当加上某一段连续负数序列后,我们血量小于等于0,此时不得不调整
那么我们自然是优先把伤害最高的调到末尾,这就要求我们获取之前遇到的所有负数中的最小值
我们开一个小根堆,存储所有遇到的负数,当我们血量小于等于0时,我们就不断弹出堆顶,并恢复相应血量,直到血量为正值
时间复杂度:O(nlogn) 空间复杂度:O(n)
class Solution {
public:
int magicTower(vector& nums) {
long long s = accumulate(nums.begin(), nums.end(), 1LL);
if(s <= 0) return -1;
priority_queue, greater> pq;
s = 1;
int ret = 0;
for(auto x : nums){
if(x < 0) pq.emplace(x);
s += x;
while(s <= 0)
s -= pq.top(), pq.pop(), ret++;
}
return ret;
}
};