NC50528 滑动窗口

滑动窗口

https://ac.nowcoder.com/acm/problem/50528

题意

给一个长度为N的数组,一个长为K的滑动窗体从最左端移至最右端,你只能看到窗口中的K个数,找出窗口中的最大值和最小值,然后将窗口往后移动一位。

思路

找最大值和找最小值是一样的方法,这里以找最大值为例。
每次窗口向右移动时,如果进来了一个比上一个值要大的数,则上一个值从此以后绝不可能成为窗口中的最大值,也就是说,当窗口的右端点从移动到时,如果,那么再也不可能是窗口中的最大值
同时,因为窗口的移动,窗口左端点上的数会离开窗口,不再被我们考虑。

这样,我们可以考虑使用双向队列,每次移动窗口时从后往前删除不再可能成为最大值的数,并且判断最前面的数是否已经离开窗口
为了判断最前面的数是否已经离开窗口,我们可以在队列中不直接储存数的大小,而是储存数的下标,每次要比较的时候再用过下标找到数,这样会方便一些。

找最小值同理,判断不可能的数的时候把符号变一下就行了

代码

#include <iostream>
#include <deque>
#include <vector>

using namespace std;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);

    int N;  //数组长度N
    int K;  //窗口长度K
    cin >> N >> K;
    int num[N + 5];
    for (int i = 1; i <= N; ++i) {
        cin >> num[i];
    }

    deque<int> maxIndex;  //记录当前窗口中最大值的下标
    deque<int> minIndex;  //记录当前窗口中最小值的下标
    vector<int> max;    //记录所有最大值
    vector<int> min;    //记录所有最小值
    maxIndex.push_back(1);
    minIndex.push_back(1);

    for (int i = 2; i <= N; ++i) {
        //维护最小值
        //如果新来的数比队列最后一个数小,把队列中最后一个数删掉
        while (!minIndex.empty() && num[i] < num[minIndex.back()]) {
            minIndex.pop_back();
        }
        //往队列里压入新来的数
        minIndex.push_back(i);
        //窗口移动后,队列最前面在窗口范围之外的数删掉
        while (minIndex.front() <= i - K) {
            minIndex.pop_front();
        }
        if (i >= K) {
            min.push_back(minIndex.front());
        }

        //维护最大值
        while (!maxIndex.empty() && num[i] > num[maxIndex.back()]) {
            maxIndex.pop_back();
        }
        maxIndex.push_back(i);
        while (maxIndex.front() <= i - K) {
            maxIndex.pop_front();
        }
        if (i >= K) {
            max.push_back(maxIndex.front());
        }
    }
    for (int &i:min) {
        cout << num[i] << " ";
    }
    cout << endl;
    for (int &i:max) {
        cout << num[i] << " ";
    }
    return 0;
}
全部评论

相关推荐

点赞 1 评论
分享
牛客网
牛客企业服务