题目描述:
有一个长为 n 的序列 a,以及一个大小为 k 的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。
题目分析:
这道题是一个滑动窗口最小最大值的算法实现。:
-
首先我们定义了一个结构体,包含两个整型变量。
-
依次处理数组中的元素,维护双端队列中的元素,确保队列中的元素是滑动窗口内的元素,并且队列中的元素按照一定的规则排列。具体做法是:
- 遍历数组,保证队列中的元素是当前滑动窗口内的元素,如果队列不为空且队首元素已经超出窗口范围,则将队首元素出队。
- 如果当前元素比队尾元素小(对于最小值的情况),或者比队尾元素大(对于最大值的情况),则将队尾元素出队,直到队列为空或者队尾元素满足条件为止。
- 将当前元素加入队尾。
- 记录当前窗口的最小值和最大值,并输出。
-
最后输出得到的最小值和最大值数组。
题目代码 :
#include<iostream>
#include<queue>
using namespace std;
struct ss {
int n;
int m;
};
deque<ss> st;
struct ss a[1000005];
int i, j, n, k;
int max1[1000005];
int min1[1000005];
int main() {
cin >> n >> k;
for (i = 1; i <= n; i++) {
cin >> a[i].n;
a[i].m = i;
}
for (i = 1; i <= n; i++) {
while (!st.empty() && st.front().m <= i - k) {
st.pop_front();
}
while (!st.empty() && st.back().n <= a[i].n) {
st.pop_back();
}
st.push_back(a[i]);
max1[i] = st.front().n;
}
st.clear();
for (i = 1; i <= n; i++) {
while (!st.empty() && st.front().m <= i - k) {
st.pop_front();
}
while (!st.empty() && st.back().n >= a[i].n) {
st.pop_back();
}
st.push_back(a[i]);
min1[i] = st.front().n;
}
for (i = k; i <= n; i++) {
cout << min1[i] << " ";
}
cout << endl;
for (i = k; i <= n; i++) {
cout << max1[i] << " ";
}
return 0;
}