虽然这个题提示了是单调队列的板子题,但是之前都没了解过单调队列这个数据结构,我还是下意识的暴力了一波,果不其然,TLE在最后一个样例上
暴力解法复杂度:O(nk)
单调队列解法复杂度: O(n) 差距还是很致命的
单调队列
百度百科给出的解释:单调队列,即单调递减或单调递增的队列。使用频率不高,但在有些程序中会有非同寻常的作用。
为了维护这个单调性,在每次往队列中插入数据时,需要注意的就是,当前待插入到队尾的值是否破坏了当前队列的单调性,即如果是维护的一个单调递增的队列(队首元素始终维护最小值),那么当待插入的数比当前队尾的数小的时候,我们需要将队尾的元素弹出,直到满足:待插入的元素比队尾的元素大的时候再将其插入(这以操作就保证了整个队列的单调性,且不破坏原有数据的相对位置,单调递减的队列原理相同,条件相反)。
由于传统的队列只能做到队尾进元素,队首出元素,但是要实现上述功能,就需要一种能够实现在队尾也能弹出元素的数据结构,因此我们引入双端队列(deque)。允许两端弹出,只允许一端插入的队列(允许两端插入,只允许一端弹出的也属于双端队列),刚好,在STL中前人已经为我们写好了这种数据结构,使用时需要引入deque这个头文件
关于deque,详细的使用方法参见博客:点击进入
具体思路
为了严格的维护队列的单调性,我们采取将原序列数对应的下标放入队列中,维护原则按照前面所讲的方法,这样一来,我们就确保了队首的下标对应的数一定是某个区间的最小值(或最大值),但是这样维护的的值可能不是我们需要计算的对应值。
对于本题中,由于我们需要计算的是在一个数组中[i-k+1,i ](1<=i<=n)区间的最小值和最大值,需要注意的是,当i小于输入的k的时候,即i到数组第一个元素小于要求的区间长度的时候,此时队列中的元素还不一定是最小(或最大)的值,我们继续将数组中的元素放入队列中,当i>=k的时候,就需要对在队首判断是否进行弹出操作,以输出最小值(或最大值),弹出的条件是当前队列中的x是否小于所求区间的左端点,小于则弹出,大于等于则输出该下表对应的数。
AC代码
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <string>
#include <math.h>
#include <set>
#include <map>
#include <vector>
#include <queue>
#include <deque>
#include <algorithm>
using namespace std;
const int maxn = 1e6 + 5;
typedef long long ll;
const int INF = 0x3f3f3f3f;
int a[maxn];
deque<int> q;
int main()
{
int n,k;
cin >> n >> k;
for(int i = 1; i <= n; i++)
cin >> a[i];
for(int i = 1; i <= n; i++){ //计算最小值
while(!q.empty() && a[q.back()] >= a[i])
q.pop_back(); //删除队尾较大的元素以维护单调性
q.push_back(i);
if(i >= k){
while(!q.empty() && q.front() < i-k+1) //这里为什么是i-k+1?
q.pop_front(); //我们把计算的区间理解成是一个左闭右闭的区间
cout << a[q.front()] << " "; // 例如 7 6 8 12 9 10 3 当i=6时 区间的开始应该是3
//即当k = 4,i = 5的时候 对应的区间应该是 [5-4+1,5]
//当队列中的i不在这个区间中时,说明队头的下标对应的值不是当前这个区间的最小值
}
}
cout << endl;
while(!q.empty()) q.pop_back();
for(int i = 1; i <= n; i++){ //计算最大值
while(!q.empty() && a[q.back()] <= a[i])
q.pop_back();
q.push_back(i);
if(i >= k){
while(!q.empty() && q.front() < i-k+1)
q.pop_front();
cout << a[q.front()] << " ";
}
}
cout << endl;
return 0;
}