单调队列-滑动窗口最优解

154. 滑动窗口 - AcWing题库

 左边是原数组，红框是画图窗口、右边是队列。这里我们采用队列存放值，而不存放原数组下标。

求滑动窗口中的最小值：当窗口中左边的数大于右边，那么她永远不可能作为最小值输出，直接从队列里移除（先进先出）。

而后进来的数比队尾的数小，则剔除队尾继续比较，直到该队列为”单调递增“位置。将每次滑动窗口最小的数放在队头，每次取最小值时取对头就行。最大值同理。

import java.io.*;

public class Main{
    static int N = 1000010;
    static int a[] = new int[N]; //原数组
    static int q[] = new int[N]; //单调增队列（队头取最小）
    static int r[] = new int[N]; //单调减队列（队头取最大）
    static int head, end; //单调增队列指针
    static int rHead,rEnd;

    static void init(){
        head = 0;
        end = 0;
        rHead = 0;
        rEnd = 0;
    }
    public static void main(String []args)throws IOException{
        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter log = new BufferedWriter(new OutputStreamWriter(System.out));
        String arr[] = in.readLine().split(" ");
        int n = Integer.parseInt(arr[0]), k =Integer.parseInt(arr[1]);
        String str[] = in.readLine().split(" ");
        init();

        for(int i = 0; i < n; i ++){
            a[i] = Integer.parseInt(str[i]);

            //判断当队列不为空、窗口已经形成并移动下一格时出队上一格最小值
            if(head != end && i >= k && a[i - k] == q[head])  head ++;
            //形成单调队列，因为尾指针指向下一个入队位置，所有这里判断用end - 1
            while(head != end && q[end - 1] > a[i]){ end --; q[end] = 0;}
            q[end++] = a[i];

            //判断已形成了窗口输出最小值，刚开始没有形成窗口
            if(i + 1>= k)  log.write(q[head]+" ");
        }
        log.write("\n");
        log.flush();

        for(int i = 0; i < n ; i ++){
            //判断当队列不为空、窗口已经形成并移动下一格时出队上一格最大值
            if(rHead != rEnd && i >= k  && a[i - k] == r[rHead]) rHead++;

            while(rHead != rEnd && r[rEnd - 1] < a[i]){ rEnd --; r[rEnd] = 0;}
            r[rEnd ++] = a[i];

            if(i + 1>= k)   log.write(r[rHead]+" ");
        }
        log.flush();
    }
}