NowCoder-[编程题]生成窗口最大值数组

NowCoder-[编程题]生成窗口最大值数组

题目

有一个整型数组arr和一个大小为w的窗口从数组的最左边滑到最右边，窗口每次向右边滑一个位置，求每一种窗口状态下的最大值。（如果数组长度为n，窗口大小为w，则一共产生n-w+1个窗口的最大值）

原题链接：https://www.nowcoder.com/questionTerminal/b316c7f9617744b98fa311ae29ac516c?f=discussion

思路

• 这道题本质上是滑动窗口问题 关键在于如何定义更新滑动窗口的规则

• 滑动窗口更新规则：
  * 首先如果双端队列是空的 直接入队 队尾
  * 如果双端队列不是空的 取出当前队尾元素的下标 记为j
  * 如果arr[i] >= arr[j] 直接将下标为j的元素弹出 然后下标为i的元素入队
  * 如果arr[i] < arr[j] 直接将下标为j的元素入队

代码

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedList;

public class Main {
    // 这道题本质上是滑动窗口问题  关键在于如何定义更新滑动窗口的规则

    public static int[] getMaxWindow(int[] arr,int w)
    {
        // 滑动窗口更新规则：
        // 首先如果双端队列是空的  直接入队 队尾
        // 如果双端队列不是空的  取出当前队尾元素的下标  记为j
        // 如果arr[i] >= arr[j]  直接将下标为j的元素弹出  然后下标为i的元素入队
        // 如果arr[i] < arr[j]   直接将下标为j的元素入队

        // 首先考虑程序的鲁棒性
        if(arr == null || w < 1 || arr.length < w)
        {
            return null;
        }
        // 创建一个双端队列
        LinkedList<Integer>  qmax = new LinkedList<Integer>();
        // 创建结果存储数组
        int[] result = new int[arr.length - w + 1];
        int index = 0;
        for(int i = 0; i < arr.length; i++)
        {
            // 如果arr[j] <= arr[i]  直接pop出arr[j]
            while(!qmax.isEmpty() && arr[i] >= arr[qmax.peekLast()])
            {
                qmax.pollLast();// 弹出队尾元素
            }

            // 此时的情况就是  arr[j] >= arr[i]  直接入队即可
            qmax.addLast(i);// 下标I入队

            // 此时判断窗口大小是否是w
            if(qmax.peekFirst() == i - w)
            {
                // 将过期的元素 弹出队
                qmax.pollFirst();
            }

            if(i >= w - 1)
            {
                // 如果窗口出现了  记录每一个窗口的最大值
                result[index++] = arr[qmax.peekFirst()];// 队头元素 就是此时的最大值
                // 维护的队列是一个递减的数列
            }
        }
        return result;
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n;
        int w;
        String line;

        while((line = br.readLine()) != null)
        {
            // 对读入的字符串进行处理
            n = Integer.parseInt(line.split(" ")[0]);
            w = Integer.parseInt(line.split(" ")[1]);
            int[] data = new int[n];
            String[] inputHelp = br.readLine().split(" ");

            for(int i = 0; i < n; i++)
            {
                data[i] = Integer.parseInt(inputHelp[i]);
            }

            int[] res = getMaxWindow(data,w);

            StringBuilder sb = new StringBuilder();

            for(int i :res)
            {
                sb.append(i).append(" ");
            }
            System.out.println(sb);
        }



    }


}