LeetCode每日一题 | LCP 30. 魔塔游戏
文章目录
- 题目描述
- 问题分析
- 程序代码
题目描述
原题链接
小扣当前位于魔塔游戏第一层，共有 N 个房间，编号为 0 ~ N-1。每个房间的补血道具/怪物对于血量影响记于数组 nums，其中正数表示道具补血数值，即血量增加对应数值；负数表示怪物造成伤害值，即血量减少对应数值；0 表示房间对血量无影响。
小扣初始血量为 1，且无上限。假定小扣原计划按房间编号升序访问所有房间补血/打怪，为保证血量始终为正值，小扣需对房间访问顺序进行调整，每次仅能将一个怪物房间（负数的房间）调整至访问顺序末尾。请返回小扣最少需要调整几次，才能顺利访问所有房间。若调整顺序也无法访问完全部房间，请返回 -1。
问题分析
当访问到第 i 个房间，当前的血量值不为正数，说明需要将一个怪物房间（负数的房间）调整至访问顺序末尾。
而为了确保调整次数最少，则需要用一个小顶堆，维护当前数值最小的怪物房间，并将其调整至顺序末尾。
程序代码
class Solution {
public:
int magicTower(vector<int>& nums) {
priority_queue<int, vector<int>, greater<int>> q;
int cnt = 0;
long long sum = 1, t = 0;
for(auto num : nums) {
if(num < 0) q.push(num);
sum += num;
if( sum <= 0 ) {
cnt++;
t += q.top();
sum -= q.top();
q.pop();
}
}
sum += t;
return sum <= 0 ? -1 : cnt;
}
};