编程问题:
小扣当前位于魔塔游戏第一层,共有 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.贪心+优先队列
计算一遍所有回合之后的血量,如果是非正数,则直接返回-1;然后模拟过程,将之前扣减的血量都放入优先队列中,每次快死之前,就取出堆顶的元素(扣最多的血)给自己加上,这样的贪心思想能保证我们移动到尾部的元素是最少的。
时间复杂度 O(NlogN)
空间复杂度 O(N)
class Solution {
public:
int magicTower(vector<int>& nums) {
long long sum = 1;
for (int x : nums)
sum += x;
if (sum <= 0)
return -1;
priority_queue<int> que;
long long blood = 1;
int cnt = 0;
for (int x : nums)
{
if (x < 0)
{
que.push(-x);
if (blood + x <= 0)
{
int cur = que.top();
que.pop();
blood += cur;
cnt++;
}
}
blood += x;
}
return cnt;
}
};