滑动窗口
题目描述
有一个长为 n 的序列 a,以及一个大小为 k 的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。
例如:
The array is [1,3,-1,-3,5,3,6,7], and k = 3。
输入格式
输入一共有两行,第一行有两个正整数 n,k。 第二行 n 个整数,表示序列 a。
输出格式
输出共两行,第一行为每次窗口滑动的最小值
第二行为每次窗口滑动的最大值
输入样例
8 3
1 3 -1 -3 5 3 6 7
输出样例
-1 -3 -3 -3 3 3
3 3 5 5 6 7
【数据范围】
对于 50% 的数据,
1
≤
n
≤
1
0
5
1 \le n \le 10^5
1≤n≤105;
对于 100% 的数据,
1
≤
k
≤
n
≤
1
0
6
1\le k \le n \le 10^6
1≤k≤n≤106,
a
i
∈
[
−
2
31
,
2
31
)
a_i \in [-2^{31},2^{31})
ai∈[−231,231)。
思路
用deque来维护一个递增的序列和一个递减的序列,每次入队时清理队头、队尾的无效元素。剩下的队头元素就是最大值、最小值了。
#include<cstdio>
#include<queue>
#include<algorithm>
using namespace std;
const int maxn=1e6+5;
struct N{int v,p;};
deque<N> big,sma;
int n,k,v,ansb[maxn],anss[maxn];
int main(){
scanf("%d%d",&n,&k);
for(int i=0;i<n;i++){
scanf("%d",&v);
while(big.size()&&big.back().v>=v) big.pop_back();
while(big.size()&&i-big.front().p>=k) big.pop_front();
big.push_back({v,i});
while(sma.size()&&sma.back().v<=v) sma.pop_back();
while(sma.size()&&i-sma.front().p>=k) sma.pop_front();
sma.push_back({v,i});
ansb[i]=sma.front().v;
anss[i]=big.front().v;
}
for(int i=k-1;i<n;i++) printf("%d%c",anss[i],i==n-1?'\n':' ');
for(int i=k-1;i<n;i++) printf("%d%c",ansb[i],i==n-1?'\n':' ');
}