POJ 滑动窗口(优先队列的应用)

数据结构与算法A (第三章 栈与队列 练习题): 滑动窗口

思路

对于最大最小值分别维护一个优先队列（保存元素下标）。以最小值为例。
每次遇到一个新元素，从队尾插入。插入时删去队列中比该值大的元素。（因为当前值出现的下标较晚，所以以后一定范围窗口的最小值不会超过该值）。队首是当前窗口的最小值。同时要注意维护队首的下标是否在当前窗口内。

#include <iostream>
#include <algorithm>
#include <string>
#include <cstring>
#include <cstdio>
#include <vector>
#include <queue>

using namespace std;

int n, k, mmin, mmax;
int arr[1010000];
// priority_queue<int> max_q;
// priority_queue<int, vector<int>, greater<int> >min_q;        // 存放当前窗口最大最小值
int min_q[1010000], max_q[1010000], min_left=0, min_right=0, max_left=0, max_right=0;
vector<int> min_v;
vector<int> max_v;      // 存放最小值和最大值序列

int main()
{
    //freopen("in.txt", "r", stdin);

    scanf("%d%d", &n, &k);
    for(int i=0;i<n;i++)
        scanf("%d", &arr[i]);

    for(int i=0;i<n;i++)
    {
        // 处理最小值队列
        // 当前处理第 i 个元素。从后方插入，遇到比他大的就删掉。
        while(min_right>min_left && arr[min_q[min_right-1]]>arr[i])
        {
            min_right--;
        }
        min_q[min_right++] = i;
        // 更新左边下标在范围之外的元素
        while(min_q[min_left] < i-k+1)
            min_left++;

        // 处理最大值队列
        while(max_right>max_left && arr[max_q[max_right-1]]<arr[i])
        {
            max_right--;
        }
        max_q[max_right++] = i;
        // 更新左边下标在范围之外的元素
        while(max_q[max_left] < i-k+1)
            max_left++;

        if(i >= k-1)
        {
            min_v.push_back(arr[min_q[min_left]]);
            max_v.push_back(arr[max_q[max_left]]);
        }
    }

    // 初始化窗口

    for(int i=0;i<n-k+1;i++)
    {
        printf("%d", min_v[i]);
        if(i != n-k)
            printf(" ");
    }
    printf("\n");
    for(int i=0;i<n-k+1;i++)
    {
        printf("%d", max_v[i]);
        if(i != n-k)
            printf(" ");
    }
    return 0;
}