生成窗口最大值数组

https://www.nowcoder.com/practice/b316c7f9617744b98fa311ae29ac516ctpId=101&tqId=33083&tPage=1&rp=1&ru=/ta/programmer-code-interview-guide&qru=/ta/programmer-code-interview-guide/question-ranking

题目描述

有一个整型数组arr和一个大小为w的窗口从数组的最左边滑到最右边，窗口每次向右边滑一个位置，求每一种窗口状态下的最大值。（如果数组长度为n，窗口大小为w，则一共产生n-w+1个窗口的最大值）

输入描述:

第一行输入n和w，分别代表数组长度和窗口大小

第二行输入n个整数Xi​，表示数组中的各个元素

输出描述:

输出一个长度为n-w+1的数组res，res[i]表示每一种窗口状态下的最大值

示例1

输入

8 3
4 3 5 4 3 3 6 7

输出

5 5 5 4 6 7

备注:

1≤w≤n≤1000000     −1000000≤Xi​≤1000000

 

算法思路：

进队列的策略：

1、如果dq为空，则直接将下标放入队列

2、如果dq非空，则比较data[i]和队列中的队尾j对应的data[j]的大小。

　　　　　　　　   若data[j]>data[i]，那么表示data[i]有可能在接下来成为最大值，所以将i放入队列

　　　　　　　　   若data[j]<=data[i]，那么表示data[j]已经不可能是最大的值了，所以将data[j]抛出，继续进行比较。

出队列的策略：

如果此时队首的下标k（ dq.front() ）满足k==i-w，则表示当前的队首元素已经过期，所以可以将队首的元素抛出。

//使用双端队列维护窗口最大值的角标
#include<bits/stdc++.h>
using namespace std;
int main(){
    int n,w;
    cin>>n>>w;
    vector<int> data(n);
    for(int i=0;i<n;i++){
        cin>>data[i];
    }
    deque<int> dq;
    vector<int> res(n-w+1);
    int index=0;
    for(int i=0;i<n;i++){
        while(!dq.empty() && data[dq.back()]<=data[i]){
            dq.pop_back();
        }
        dq.push_back(i);
        if(dq.front()==i-w){
            dq.pop_front();
        }
        if(i>=w-1){  //从该位置处，res中开始不断存储当前窗口的最大值
            res[index++]=data[dq.front()];
        }
    }
    for(auto x : res){
        cout<<x<<" ";
    }
    return 0;
}