【数组】剑指Offer: 滑动窗口的最大值

【在线编程】滑动窗口的最大值

【问题描述】
给定一个数组和滑动窗口的大小，找出所有滑动窗口里数值的最大值。例如，如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3，那么一共存在6个滑动窗口，他们的最大值分别为{4,4,6,6,6,5}； 针对数组{2,3,4,2,6,2,5,1}的滑动窗口有以下6个： {[2,3,4],2,6,2,5,1}， {2,[3,4,2],6,2,5,1}， {2,3,[4,2,6],2,5,1}， {2,3,4,[2,6,2],5,1}， {2,3,4,2,[6,2,5],1}， {2,3,4,2,6,[2,5,1]}。

【解题思路 & Java实现】

方法一：暴力解决。时间复杂度较高,为O(nk),k为滑动窗口大小。

import java.util.ArrayList;

public class Solution {
    public ArrayList<Integer> maxInWindows(int[] num, int size) {
        ArrayList<Integer> result = new ArrayList<Integer>();

        if (num == null || num.length == 0 || size > num.length || size == 0) {
            return result;
        }

        for (int i = 0; i <= num.length - size; i++) {
            int max = num[i];
            for (int j = i + 1; j < i + size; j++) {
                if (num[j] > max) {
                    max = num[j];
                }
            }
            result.add(max);
        }
        return result;
    }
}

方法二：使用一个双端队列，队列的首部存放当前滑动窗口的最大值，队列的尾部依次存放可能为下个滑动窗口的最大值。每当窗口滑动一次需要判断：

1. 新增加的元素从队尾开始比较，把比它小的元素全部删除；
2. 队列中当前最大元素的下标是否还在新的滑动窗口内。
   时间复杂度O(n)，空间复杂度O(k).

import java.util.ArrayList;
import java.util.LinkedList;

public class Solution {
    public ArrayList<Integer> maxInWindows(int[] num, int size) {
        ArrayList<Integer> result = new ArrayList<>();
        if (num == null || num.length == 0 || size <= 0 || num.length < size) {
            return result;
        }

        //双端队列，用来记录每个窗口的最大值下标
        LinkedList<Integer> index = new LinkedList<>();

        for (int i = 0; i < num.length; i++) {
            //把队列中比 num[i]小的元素全部删除
            while (!index.isEmpty() && num[index.peekLast()] < num[i]) {
                index.pollLast();
            }
            index.addLast(i);

            //判断队首元素（最大元素）是否在新的滑动窗口内
            if (index.peekFirst() <= i - size) {
                index.pollFirst();
            }
            //向result列表中加入元素，注意当 i 遍历了size个才开始往 result中添加最大元素
            if (i >= size - 1) {
                result.add(num[index.peekFirst()]);
            }
        }
        return result;
    }
}