昨晚刚看CS-Notes的容器部分知道了Java还有个PriorityQueue(优先队列),没想到今天 的leetCode周赛就果断出了相关题,依旧喜闻乐见不会做。
解题方法:
定义左右指针指向当前滑动窗口的两端
int left = 0;
int right = 0;
然后定义两个优先队列,一个单调递增,一个单调递减以获取当前窗口的最大最小值
//优先队列,自然增长从小到大
PriorityQueue<Integer> minQueue = new PriorityQueue<>(Comparator.naturalOrder());
//优先队列,自然增长从大到小
PriorityQueue<Integer> maxQueue = new PriorityQueue<>(Comparator.reverseOrder());
然后进行逻辑判断,当minQueue和maxQueue的队首差值在limite限制内,只需要窗口向右滑动,如果不在,需要窗口整体向右滑动一位,并且left的值需要在两个队列中剔除掉。
最后代码如下:
public int longestSubarray(int[] nums, int limit) {
//优先队列,自然增长从小到大
PriorityQueue<Integer> minQueue = new PriorityQueue<>(Comparator.naturalOrder());
//优先队列,自然增长从大到小
PriorityQueue<Integer> maxQueue = new PriorityQueue<>(Comparator.reverseOrder());
int left = 0;
int right = 0;
int count = 0;
while (left<nums.length && right<nums.length){
minQueue.add(nums[right]);
maxQueue.add(nums[right]);
if(maxQueue.peek()-minQueue.peek()<=limit){
count = Math.max(count,right-left+1);
right++;
continue;
}
//建议如果是为了删除值的话,先进性装箱操作,虽然在PriorityQueue不会有问题,但是在ArrayList方法中remove有两种重载一种是remove(int)删除索引为int的值,一种是remove(object)删除值
minQueue.remove((Integer)nums[left]);
maxQueue.remove((Integer)nums[left]);
left++;
right++;
}
return count;
}