前文用 单调栈解决三道算法问题 介绍了单调栈这种特殊数据结构,本文写一个类似的数据结构「单调队列」。

也许这种数据结构的名字你没听过,其实没啥难的,就是一个「队列」,只是使用了一点巧妙的方法,使得队列中的元素全都是单调递增(或递减)的

「单调栈」主要解决 Next Great Number 一类算法问题,而「单调队列」这个数据结构可以解决滑动窗口问题。之前的 滑动窗口解题套路框架 讲的滑动窗口算法是双指针技巧的一种,是解决子串、子数组的通用技巧;而本文说的滑动窗口是比较具体的问题。

比如说力扣第 239 题「滑动窗口最大值」,难度 Hard

给你输入一个数组nums和一个正整数k,有一个大小为k的窗口在nums上从左至右滑动,请你输出每次窗口中k个元素的最大值。

函数签名如下:

  1. int[] maxSlidingWindow(int[] nums, int k);

比如说题目给出的一个示例:

单调队列 - 图1

一、搭建解题框架

这道题不复杂,难点在于如何在O(1)时间算出每个「窗口」中的最大值,使得整个算法在线性时间完成。这种问题的一个特殊点在于,「窗口」是不断滑动的,也就是你得动态地计算窗口中的最大值。

对于这种动态的场景,很容易得到一个结论:

在一堆数字中,已知最值为**A**,如果给这堆数添加一个数**B**,那么比较一下**A****B**就可以立即算出新的最值;但如果减少一个数,就不能直接得到最值了,因为如果减少的这个数恰好是**A**,就需要遍历所有数重新找新的最值

回到这道题的场景,每个窗口前进的时候,要添加一个数同时减少一个数,所以想在 O(1) 的时间得出新的最值,不是那么容易的,需要「单调队列」这种特殊的数据结构来辅助。

一个普通的队列一定有这两个操作:

  1. class Queue {
  2. // enqueue 操作,在队尾加入元素 n
  3. void push(int n);
  4. // dequeue 操作,删除队头元素
  5. void pop();
  6. }

一个「单调队列」的操作也差不多:

  1. class MonotonicQueue {
  2. // 在队尾添加元素 n
  3. void push(int n);
  4. // 返回当前队列中的最大值
  5. int max();
  6. // 队头元素如果是 n,删除它
  7. void pop(int n);
  8. }

当然,这几个 API 的实现方法肯定跟一般的 Queue 不一样,不过我们暂且不管,而且认为这几个操作的时间复杂度都是 O(1),先把这道「滑动窗口」问题的解答框架搭出来:

  1. int[] maxSlidingWindow(int[] nums, int k) {
  2. MonotonicQueue window = new MonotonicQueue();
  3. List<Integer> res = new ArrayList<>();
  4. for (int i = 0; i < nums.length; i++) {
  5. if (i < k - 1) {
  6. //先把窗口的前 k - 1 填满
  7. window.push(nums[i]);
  8. } else {
  9. // 窗口开始向前滑动
  10. // 移入新元素
  11. window.push(nums[i]);
  12. // 将当前窗口中的最大元素记入结果
  13. res.add(window.max());
  14. // 移出最后的元素
  15. window.pop(nums[i - k + 1]);
  16. }
  17. }
  18. // 将 List 类型转化成 int[] 数组作为返回值
  19. int[] arr = new int[res.size()];
  20. for (int i = 0; i < res.size(); i++) {
  21. arr[i] = res.get(i);
  22. }
  23. return arr;
  24. }

单调队列 - 图2

这个思路很简单,能理解吧?下面我们开始重头戏,单调队列的实现。

二、实现单调队列数据结构

观察滑动窗口的过程就能发现,实现「单调队列」必须使用一种数据结构支持在头部和尾部进行插入和删除,很明显双链表是满足这个条件的。

「单调队列」的核心思路和「单调栈」类似,push方法依然在队尾添加元素,但是要把前面比自己小的元素都删掉:

  1. class MonotonicQueue {
  2. // 双链表,支持头部和尾部增删元素
  3. private LinkedList<Integer> q = new LinkedList<>();
  4. public void push(int n) {
  5. // 将前面小于自己的元素都删除
  6. while (!q.isEmpty() && q.getLast() < n) {
  7. q.pollLast();
  8. }
  9. q.addLast(n);
  10. }
  11. }

你可以想象,加入数字的大小代表人的体重,把前面体重不足的都压扁了,直到遇到更大的量级才停住。

单调队列 - 图3

如果每个元素被加入时都这样操作,最终单调队列中的元素大小就会保持一个单调递减的顺序,因此我们的max方法可以可以这样写:

  1. public int max() {
  2. // 队头的元素肯定是最大的
  3. return q.getFirst();
  4. }

pop方法在队头删除元素n,也很好写:

  1. public void pop(int n) {
  2. if (n == q.getFirst()) {
  3. q.pollFirst();
  4. }
  5. }

之所以要判断data.front() == n,是因为我们想删除的队头元素n可能已经被「压扁」了,可能已经不存在了,所以这时候就不用删除了:

单调队列 - 图4

至此,单调队列设计完毕,看下完整的解题代码:

/* 单调队列的实现 */
class MonotonicQueue {
    LinkedList<Integer> q = new LinkedList<>();
    public void push(int n) {
        // 将小于 n 的元素全部删除
        while (!q.isEmpty() && q.getLast() < n) {
            q.pollLast();
        }
        // 然后将 n 加入尾部
        q.addLast(n);
    }

    public int max() {
        return q.getFirst();
    }

    public void pop(int n) {
        if (n == q.getFirst()) {
            q.pollFirst();
        }
    }
}

/* 解题函数的实现 */
int[] maxSlidingWindow(int[] nums, int k) {
    MonotonicQueue window = new MonotonicQueue();
    List<Integer> res = new ArrayList<>();

    for (int i = 0; i < nums.length; i++) {
        if (i < k - 1) {
            //先填满窗口的前 k - 1
            window.push(nums[i]);
        } else {
            // 窗口向前滑动,加入新数字
            window.push(nums[i]);
            // 记录当前窗口的最大值
            res.add(window.max());
            // 移出旧数字
            window.pop(nums[i - k + 1]);
        }
    }
    // 需要转成 int[] 数组再返回
    int[] arr = new int[res.size()];
    for (int i = 0; i < res.size(); i++) {
        arr[i] = res.get(i);
    }
    return arr;
}

有一点细节问题不要忽略,在实现MonotonicQueue时,我们使用了 Java 的LinkedList,因为链表结构支持在头部和尾部快速增删元素;而在解法代码中的res则使用的ArrayList结构,因为后续会按照索引取元素,所以数组结构更合适。

c++代码

class MonotonicQueue{
private:
    list<int> queue;
public:
    void push(int n){
        // 将小于 n 的元素全部删除
        while(!queue.empty()&& queue.back()<n) queue.pop_back();
        queue.push_back(n);
    }
    int max(){
        return queue.front();
    }
    void pop(int n){
        if(!queue.empty()&& n==queue.front())
            queue.pop_front();
    }
};

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        MonotonicQueue window;
        vector<int> res;
        for(int i=0;i<nums.size();++i){
            if(i<k-1)window.push(nums[i]);//先填满窗口的前 k - 1
            else{
                window.push(nums[i]);// 窗口向前滑动,加入新数字
                res.push_back(window.max());// 记录当前窗口的最大值
                window.pop(nums[i-k+1]);// 移出旧数字
            }
        }
        return res;
    }
};

三、算法复杂度分析

读者可能疑惑,push操作中含有 while 循环,时间复杂度应该不是O(1)呀,那么本算法的时间复杂度应该不是线性时间吧?

单独看push操作的复杂度确实不是O(1),但是算法整体的复杂度依然是O(N)线性时间。要这样想,nums中的每个元素最多被push_backpop_back一次,没有任何多余操作,所以整体的复杂度还是O(N)

空间复杂度就很简单了,就是窗口的大小O(k)