双端队列，生成窗口最大值数组！

微信公众号：小林玩大数据
如果你觉得本文对你有帮助，拉到文末点个在看

题目描述

解题思路

代码实现

题目描述

【生成窗口最大值数组】
有一个整形数组 arr，和一个大小为 w 的窗口从最左边滑到最右边，窗口每次向右边滑动一个位置。

例如，数组为[4,3,5,4,3,3,6,7]，窗口大小为3
[4 3 5] 4 3 3 6 7 窗口最大值为5
4 [3 5 4] 3 3 6 7 窗口最大值为5
4 3 [5 4 3] 3 6 7 窗口最大值为5
4 3 5 [4 3 3] 6 7 窗口最大值为4
4 3 5 4 [3 3 6] 7 窗口最大值为6
4 3 5 4 3 [3 6 7] 窗口最大值为7

设计一个函数:

• 输入：整形数组 arr，窗口大小为 w。
• 输出：一个长度为 n-w+1 的数组res，res[i]表示每一种窗口下的最大值。

 

解题思路

思路：利用双端队列，来实现窗口的最大值的更新。首先生成双端队列 qmax ，存放数组 arr 的下标。

• 假如遍历到arr[i],放入规则如下：
  1.当 qmax 为空时，直接把下标 i 放入qmax，放入过程结束
  2.如果 qmax 不为空时，取出当前 qmax 队尾存放的下标，假设为 j
  3.arr[j] > arr[i],直接把下标 i 放入qmax ，放入过程结束
  4.arr[j] <= arr[i] ,把下标 j 从qmax 取出，继续放入规则

• 假如遍历到arr[i],取出规则如下：
  如果 i = i-w 时，则认为下标过期，即下标不属于这个窗口内的下标，把其弹出。

根据如上的放入和弹出规则，qmax 便成了一个维护窗口为 w 的子数组的最大值更新的结构。

代码实现

 1package com.bjsxt.stackandqueue;
 2
 3import java.util.LinkedList;
 4
 5/**
 6 * 设计一个函数：
 7 * 输入：大小为 n 的整型数组和一个大小为 w 的窗口
 8 * 输出：返回一个 n-w+1 数组res[]，res[i] 代表每一种窗口状态下的最大值，
 9 */
10
11public class GetMaxWindowValue {
12    public static void main(String[] args) {
13        int [] b = {4,3,5,4,3,3,6,7};
14        int [] res = new int[b.length-3+1];
15        res = getMaxWindow(b,3);
16        for (int i=0;i<res.length;i++) {
17            System.out.println(res[i]);
18        }
19    }
20
21    public static int[] getMaxWindow(int [] arr,int w) {
22        if (arr == null || w < 0 || arr.length < w) {
23            return null;
24        }
25        //申请存放窗口最大值的数组，大小为n-w+1
26        int [] res = new int[arr.length-w+1];
27        LinkedList<Integer> qmax = new LinkedList<Integer>();
28        int index = 0;
29        for (int i = 0; i < arr.length; i++) {
30            while (!qmax.isEmpty() && arr[qmax.peekLast()] <= arr[i]) {
31                qmax.pollLast();
32            }
33            qmax.addLast(i);
34            if (qmax.peekFirst() == i-w) {
35                qmax.pollFirst();
36            }
37            if (i >= w-1) {
38                res[index++] = arr[qmax.peekFirst()];
39            }
40        }
41        return res;
42    }
43}

运行结果：

15
25
35
44
56
67

因为需要遍历数组，所以时间复杂度为0（n），空间复杂度为0(n)