NC 51001 STL / 单调队列

题意

传送门 NC 51001

题解

STL

$multiset$ 维护有序的窗口元素。

#include <bits/stdc++.h>
using namespace std;
#define maxn 1000005
int n, k;
int a[maxn], mn[maxn], mx[maxn];

int main()
{
    scanf("%d%d", &n, &k);
    for (int i = 0; i < n; i++) scanf("%d", a + i);
    multiset<int> st;
    int j = 0;
    for (int i = 0; i < n; i++)
    {
        st.insert(a[i]);
        if (i < k - 1) continue;
        if (i - k >= 0)
        {
            multiset<int>::iterator it = st.find(a[i - k]);
            st.erase(it);
        }
        mn[j] = *st.begin(), mx[j] = *(--st.end());
        j++;
    }
    for (int i = 0; i < j; i++) printf("%d%c", mn[i], i == j - 1 ? '\n' : ' ');
    for (int i = 0; i < j; i++) printf("%d%c", mx[i], i == j - 1 ? '\n' : ' ');
    return 0;
}

单调队列

单调递增队列维护窗口最小值，单调递减队列维护窗口最大值。

#include <bits/stdc++.h>
using namespace std;
#define maxn 1000005
int n, k;
int a[maxn], q[maxn];

int main()
{
    scanf("%d%d", &n, &k);
    for (int i = 0; i < n; i++) scanf("%d", a + i);
    // min
    int l = 0, r = -1;
    for (int i = 0; i < n; i++)
    {
        while (l <= r && i - q[l] >= k) ++l;
        while (l <= r && a[q[r]] > a[i]) --r;
        q[++r] = i;
        if(i < k - 1) continue;
        printf("%d ", a[q[l]]);
    }
    putchar('\n');
    // max
    l = 0, r = -1;
    for (int i = 0; i < n; i++)
    {
        while (l <= r && i - q[l] >= k) ++l;
        while (l <= r && a[q[r]] < a[i]) --r;
        q[++r] = i;
        if(i < k - 1) continue;
        printf("%d ", a[q[l]]);
    }
    putchar('\n');
    return 0;
}