题目链接:239. 滑动窗口最大值 - 力扣(LeetCode)
作者思考:
这种题目一拿到手就应该想到暴力算法行不通,因为滑动窗口的时间复杂度就是O(n)了,而且要比较出滑动窗口内的最大值,就和滑动窗口的大小k有关,那暴力算法的时间复杂度就是O(kn),且k的取值可以取到很大,这样暴力算法一定会超时。
那么我们需要一个队列,这个队列呢,放进窗口里的元素,然后随着窗口的移动,队列也一进一出,每次移动后,队列告诉我们里面的最大值是什么。那这个队列应该长这样
class MyQueue{
void add(int value) {
}
void poll(int value) {
}
int peek() {
return myQueue.peek();
}
}
每次窗口移动的时候,调用myQueue.poll(滑动窗口中移除元素的数值),myQueue.add(滑动窗口中添加元素的数值),然后myQueue.peek()直接返回当前队列中的最大值。这题题不就解决了。收工下班!!!那我们现在要找,有没有这种现成的数据结构可以拿过来直接用呢?可惜并没有,也就是说,我们需要自己实现这个队列并完成上述操作。
再分析一下这个队列我们需要给他进行大概的排序,也就是说队列的头部一定是该队列中的最大值,那我们怎么样才能将插入队列的元素中的最大值排到队列头部呢?(可以创建一个双端队列,队列的两端都可以进行出队和入队的操作)
难点:如何才能保证目前队列中的最大值在队列的头部
根据这个动画可以很形象的看出,当我们插入元素时,如果插入元素(val)大于当前队列中的元素(deque.peek())将之前的元素全部poll出去,也就是说val是队列现在的最大值,且在当前队列的出口;再加入的元素如果没有比val大,就正常插入到val后。(这就是要维护队列中最大元素)
class MyQueue{
Deque<Integer> deque = new LinkedList<>();//双端队列 两端都可以进行入队出队
void poll(int val) {
//判断当前要弹出的元素的数值是否等于队列出口的数值,如果相等则弹出
//同时判断队列是否为空
if (!deque.isEmpty() && val == deque.peek()) {
deque.poll();
}
}
//添加元素时,val大于队列中的最后一个元素,就弹出
//保证队的出口处是最大值
void add(int val) {
while (!deque.isEmpty() && val > deque.getLast()) {
deque.removeLast();
}
deque.add(val);
}
int peek() {
return deque.peek();
}
}
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
if (nums.length == 1) {
return nums;
}
int len = nums.length - k +1;//存入结果数组的长度
int[] res = new int[len];
MyQueue myQueue = new MyQueue();
int count = 0;
for (int i = 0; i < k; i++) {
myQueue.add(nums[i]);//将数组元素添加到队列中
}
res[count++] = myQueue.peek();//取出队顶的值(该队列的最大值)
for (int i = k; i < nums.length; i++) {
myQueue.poll(nums[i - k]);
myQueue.add(nums[i]);
//记录当前最大值
res[count++] = myQueue.peek();
}
return res;
}
}