Description
有一个长为 n 的序列 a,以及一个大小为 k 的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。
例如:
The array is [1,3,−1,−3,5,3,6,7] and k=3。
Input
输入一共有两行,第一行有两个正整数 n,k。
第二行 n 个整数,表示序列 a
Output
输出共两行,第一行为每次窗口滑动的最小值
第二行为每次窗口滑动的最大值
Sample 1
Inputcopy | Outputcopy |
---|---|
8 3 1 3 -1 -3 5 3 6 7 |
-1 -3 -3 -3 3 3 3 3 5 5 6 7 |
Hint
【数据范围】
对于50% 的数据,1≤n≤10^5;
对于100% 的数据,1≤k≤n≤10^6,ai∈[−231,231)。
#include<iostream>
using namespace std;
const int N = 1e6 + 10;
int a[N], maxq[N], minq[N];
int n, m;
//h是队头,用于输出答案;t是队尾,用于加入和删除元素
//注意:对头为左,队尾为右,队头只出,队尾可出可进;因此若队内有元素,则h<=t
//max队内的元素因保持单调递减的性质,加入的不是元素值,而是元素所在数组a中的下标
void maxfind() {
int h = 0, t = -1;
for (int i = 1; i <= n; i++)
{
if (h <= t && maxq[h] <= i - m) h++; //不在当前滑动窗口的元素删除
while (h <= t && a[i] >= a[maxq[t]]) t--; //a[i]为当前元素,若当前元素大于队尾的元素,则需要删除,时其满足单调递减的性质
maxq[++t] = i; //加入新元素的下标
if(i>=m) cout << a[maxq[h]] << " ";//输出结果
}
}
//原理同上
void minfind() {
int h = 0, t = -1;
for (int i = 1; i <= n; i++)
{
if (h <= t && minq[h] <= i - m) h++;
while (h <= t && a[i] <= a[minq[t]]) t--;
minq[++t] = i;
if(i>=m) cout << a[minq[h]] << " ";
}
}
int main()
{
cin >> n >> m;
for (int i = 1; i <= n; i++) cin >> a[i];
minfind();
cout << endl;
maxfind();
return 0;
}