luogu 1886 滑动窗口 (单调队列)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Tc_To_Top/article/details/82933983

题目描述

现在有一堆数字共N个数字(N<=10^6),以及一个大小为k的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。

例如:

The array is [1 3 -1 -3 5 3 6 7], and k = 3.

 

输入输出格式

输入格式:

输入一共有两行,第一行为n,k。

第二行为n个数(<INT_MAX).

输出格式:

输出共两行,第一行为每次窗口滑动的最小值

第二行为每次窗口滑动的最大值

输入输出样例

输入样例#1:

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

输出样例#1:

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

说明

50%的数据,n<=10^5

100%的数据,n<=10^6

题目链接:https://www.luogu.org/problemnew/show/P1886

题目分析:用两个单调队列维护

#include <cstdio>
#include <deque>
using namespace std;
int const MAX = 1000005;
deque<int> dqmax, dqmin;
int a[MAX], n, k;

int main() {
    scanf("%d %d", &n, &k);
    for (int i = 0; i < n; i++) {
        scanf("%d", &a[i]);
    }
    for (int i = 0; i < n; i++) {
        // 维持队列的递增性
        while (dqmin.size() > 0 && a[dqmin.back()] > a[i]) {
            dqmin.pop_back();
        }
        dqmin.push_back(i);
        // 保证在规定区间范围内
        while (dqmin.size() > 0 && dqmin.front() <= i - k) {
            dqmin.pop_front();
        }
        if (i + 1 >= k) {
            printf("%d%c", a[dqmin.front()], (i == n - 1) ? '\n' : ' ');
        }
    }
    for (int i = 0; i < n; i++) {
        while (dqmax.size() > 0 && a[dqmax.back()] < a[i]) {
            dqmax.pop_back();
        }
        dqmax.push_back(i);
        while (dqmax.size() > 0 && dqmax.front() <= i - k) {
            dqmax.pop_front();
        }
        if (i + 1 >= k) {
            printf("%d%c", a[dqmax.front()], (i == n - 1) ? '\n' : ' ');
        }
    }
}

猜你喜欢

转载自blog.csdn.net/Tc_To_Top/article/details/82933983
今日推荐