2024.2.6
题目来源
我的题解
方法一 贪心+优先队列
思路 :使用贪心的思想,从左到右遍历,若遇到加上当前房间的生命值后小于等于0,由于需要调整的次数最小,则贪心地将当前以及前面房间中生命值最小的 移到末尾。直到遍历完所有房间。
具体:在遍历房间的过程中,将为负数的生命值加入到一个小根堆pq中,当计算完每个房间的生命值sum影响后,如果生命值sum小于等于0,则将堆顶元素取出,并使用外的变量other记录从小根堆pq中取出元素的和,这时需要在生命值中补回相应的生命值以及调整次数加1。当遍历完所有房间有,再将other的值重新加入到sum中,若最终的sum小于等于0,则表示无解。
时间复杂度 :O(nlogn)。需要遍历一次数组O(n),并且遍历过程中存在优先队列的入队和出队操作O(logn)
空间复杂度:O(n)。最多所有的元素都为负数。
java
public int magicTower(int[] nums) {
//记录交换的次数
int count=0;
//记录和,有坑......可能出现整形溢出
long sum=1;
//交换到末尾的值的和
int other=0;
PriorityQueue<Integer> pq=new PriorityQueue<>();
for(int i=0;i<nums.length;i++){
int t=nums[i];
//若为负数加入到小根堆
if(t<0)
pq.offer(t);
//更新和
sum+=t;
//判断更新后的和是否小于等于0
if(sum<=0){
int temp=pq.poll();
//补回生命值
sum-=temp;
//交换到末尾的和
other+=temp;
count++;
}
}
//最终加上交换到末尾的负数和
sum+=other;
return sum>0?count:-1;
}
有任何问题,欢迎评论区交流,欢迎评论区提供其它解题思路(代码),也可以点个赞支持一下作者哈😄~