其他分享
首页 > 其他分享> > 滑动窗口最大值-队列

滑动窗口最大值-队列

作者:互联网

https://leetcode-cn.com/problems/sliding-window-maximum/

思路0:直接遍历,对比k个元素。

思路1:先确认最大值数组个数,踢出掉不能作为滑动窗口的第一个值。使用双端队列来解决问题,双端队列保存的是索引

 

public class _239_滑动窗口最大值 {

public int[] maxSlidingWindow(int[] nums, int k) {
if (nums == null || nums.length == 0 || k < 1) return new int[0];
if (k == 1) return nums;

int[] maxes = new int[nums.length - k + 1];
// 当前滑动窗口的最大值索引
int maxIdx = 0;
// 求出前k个元素的最大值索引
for (int i = 1; i < k; i++) {
if (nums[i] > nums[maxIdx]) maxIdx = i;
}

// li是滑动窗口的最左索引
for (int li = 0; li < maxes.length; li++) {
// ri是滑动窗口的最右索引
int ri = li + k - 1;
if (maxIdx < li) { // 最大值的索引不在滑动窗口的合理范围内
// 求出[li, ri]范围内最大值的索引
maxIdx = li;
for (int i = li + 1; i <= ri; i++) {
if (nums[i] > nums[maxIdx]) maxIdx = i;
}
} else if (nums[ri] >= nums[maxIdx]) { // 最大值的索引在滑动窗口的合理范围内
maxIdx = ri;
}
maxes[li] = nums[maxIdx];
}

return maxes;
}

public int[] maxSlidingWindow_deque(int[] nums, int k) {
if (nums == null || nums.length == 0 || k < 1) return new int[0];
if (k == 1) return nums;

int[] maxes = new int[nums.length - k + 1];

// peek: 取值(偷偷瞥一眼)
// poll: 删除(削)
// offer: 添加(入队)
Deque<Integer> deque = new LinkedList<>();
for (int ri = 0; ri < nums.length; ri++) {
// 只要nums[队尾] <= nums[i],就删除队尾
while (!deque.isEmpty() && nums[ri] >= nums[deque.peekLast()]) {
// deque.pollLast();
deque.removeLast();
}

// 将i加到队尾
// deque.offerLast(ri);
deque.addLast(ri);

// 检查窗口的索引是否合法
int li = ri - k + 1;
if (li < 0) continue;

// 检查队头的合法性
if (deque.peekFirst() < li) {
// 队头不合法(失效,不在滑动窗口索引范围内)
// deque.pollFirst();
deque.removeFirst();
}

// 设置窗口的最大值
maxes[li] = nums[deque.peekFirst()];
}
return maxes;
}
}

标签:maxIdx,deque,nums,队列,最大值,li,int,滑动,ri
来源: https://www.cnblogs.com/syh918/p/14939267.html