LCP 30-魔塔游戏

Raphael Liu Lv10

小扣当前位于魔塔游戏第一层,共有 N 个房间,编号为 0 ~ N-1。每个房间的补血道具/怪物对于血量影响记于数组
nums,其中正数表示道具补血数值,即血量增加对应数值;负数表示怪物造成伤害值,即血量减少对应数值;0 表示房间对血量无影响。 小扣初始血量为
1,且无上限
。假定小扣原计划按房间编号升序访问所有房间补血/打怪,为保证血量始终为正值,小扣需对房间访问顺序进行调整,每次仅能将一个怪物房间(负数的房间)调整至访问顺序末尾。请返回小扣最少需要调整几次,才能顺利访问所有房间。若调整顺序也无法访问完全部房间,请返回
-1。 示例 1: >输入:nums = [100,100,100,-250,-60,-140,-50,-50,100,150] >

输出:1 > >解释:初始血量为 1。至少需要将 nums[3] 调整至访问顺序末尾以满足要求。 示例 2: >输入:nums = [-200,-300,400,0] > >输出:-1 > >解释:调整访问顺序也无法完成全部房间的访问。 提示: - 1 <= nums.length <= 10^5 - -10^5 <= nums[i] <= 10^5

image.png

思路

首先,我们计算一遍所有回合之后的血量,如果是负数,则直接返回-1
然后我们模拟过程,将之前扣减的血量都放入优先队列中,每次快死之前,就取出堆顶的元素(扣最多的血)给自己加上,这样的贪心思想能保证我们移动到尾部的元素是最少的。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
class Solution {
public int magicTower(int[] nums) {
int sum = 1;
for(int i = 0 ; i < nums.length ; i ++){
sum += nums[i];
}
if(sum <= 0) return -1; //算出所有回合后的血量是否为正数

//开始模拟
long blood = 1;
PriorityQueue<Integer> pq = new PriorityQueue<>();
int last = 0;
for(int i = 0 ; i < nums.length ; i ++){
if(nums[i] < 0){
pq.offer(nums[i]);
if(blood + nums[i] <= 0){ //这回合过后就要死了,需要把前面扣最多的血移到最后去
last ++; //记录移动的次数
blood -= pq.poll(); //加回之前扣除最多的血量
}
}
blood += nums[i];
}
return last;
}
}

复杂度分析

时间复杂度 O(n*logn) //优先队列插入时的复杂度
空间复杂度 O(n)

 Comments
On this page
LCP 30-魔塔游戏