傅芳杰

个人博客

You Deserve The Best


单调队列

AcWing 154. 滑动窗口 原题链接

给定一个大小为n≤106n≤106的数组。

有一个大小为k的滑动窗口,它从数组的最左边移动到最右边。

您只能在窗口中看到k个数字。

每次滑动窗口向右移动一个位置。

以下是一个例子:

该数组为[1 3 -1 -3 5 3 6 7],k为3。

窗口位置 最小值 最大值
[1 3 -1] -3 5 3 6 7 -1 3
1 [3 -1 -3] 5 3 6 7 -3 3
1 3 [-1 -3 5] 3 6 7 -3 5
1 3 -1 [-3 5 3] 6 7 -3 5
1 3 -1 -3 [5 3 6] 7 3 6
1 3 -1 -3 5 [3 6 7] 3 7

您的任务是确定滑动窗口位于每个位置时,窗口中的最大值和最小值。

输入格式

输入包含两行。

第一行包含两个整数n和k,分别代表数组长度和滑动窗口的长度。

第二行有n个整数,代表数组的具体数值。

同行数据之间用空格隔开。

输出格式

输出包含两个。

第一行输出,从左至右,每个位置滑动窗口中的最小值。

第二行输出,从左至右,每个位置滑动窗口中的最大值。

输入样例:

8 3
1 3 -1 -3 5 3 6 7

输出样例:

-1 -3 -3 -3 3 3
3 3 5 5 6 7
public static void main(String[] args) {
        int n = in.nextInt();
        int k = in.nextInt();
        int[] arr = new int[n];
        in.nextIntegerArray(arr);
        int r = 0;
        List<Integer> list = new ArrayList<>();
        while (r < n) {
            if (!list.isEmpty() && r - list.get(0) + 1 > k) {
                list.remove(0);
            }
            while (!list.isEmpty() && arr[list.get(list.size() - 1)] >= arr[r]) {
                list.remove(list.size() - 1);
            }
            list.add(r);
            if (r >= k - 1)
                out.print(arr[list.get(0)] + " ");
            r++;
        }
        out.println();
        list.clear();
        r = 0;
        while (r < n) {
            if (!list.isEmpty() && r - list.get(0) + 1 > k) {
                list.remove(0);
            }
            while (!list.isEmpty() && arr[list.get(list.size() - 1)] <= arr[r]) {
                list.remove(list.size() - 1);
            }
            list.add(r);
            if (r >= k - 1)
                out.print(arr[list.get(0)] + " ");
            r++;
        }
        out.flush();
        out.close();
    }