描述:
在一个包含n个元素的数组上,有一个长度为k的窗户在从左向右滑动。窗户每滑动到一个位置,我们都可以看到k个元素在窗户中。如下的例子所示,假设数组为 [1 3 -1 -3 5 3 6 7],而k等于3: 对于窗户滑动过的每个位置,请给出窗户内k个元素的最小值和最大值。
输入
输入的第一行包括两个整数n,k,n表示数组的长度,k表示窗户的长度。 接下来一行包括n个整数,表示这个n个元素的数组。
输出
输出包含两行,每行包括n-k+1个整数,第一行表示窗户从左到右滑动过程中的最小值,第二行表示窗户从左到右滑动过程中的最大值。
样例
输入复制
8 3
1 3 -1 -3 5 3 6 7
输出复制
-1 -3 -3 -3 3 3
3 3 5 5 6 7
提示
对于100%的数据,3<=n<=1000000,1<=k<=n,数组中的每个元素均在int范围内
分析:
算法:单调队列、RMQ和线段树
但是——
还是单调队列快
RMQ和线段树会慢很多。。。
那么——
思路来了——
以最小值为例:构造一个单调队列b[i],记h为元素下标,b中存下标。第一次:b={1};第二次:b={1,2}(因为3>1且在范围内);第三次:b={3}(因为队列单调递增,-1最小,所以挤掉1和3,找最小值的时候也不会轮到它们);第四次:b={4}(1已经在外面了。-3<-1果断挤掉);第五次:b={4,5}(5>-3 可以作为候选人);第六次:b={4,6}(果断挤掉,有比5合适的候选人);第七次:b={6,7}(4不在范围内,出列);第八次:b={6,7,8}
每次都输出队列中的第一个元素。
以此维护队列中元素序号的单调性,得到结果。
只是比较判断,时间复杂度很小。
相等的值也要挤掉。
最大值同理。
............
所以——
上代码:
#include<bits/stdc++.h>
using namespace std;
#define int long long
int n,k;
vector<int> a;
deque<int> q;
deque<int> q2;
signed main()
{
ios::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
cin >> n >> k;
for(int i = 0;i < n;i++)
{
int tmp;
cin >> tmp;
a.push_back(tmp);
while(!q.empty() && a[q.back()] > a[i])
{
q.pop_back();
}
q.push_back(i);
if(!q.empty() && q.front() < (i - k + 1))
{
q.pop_front();
}
if(i >= k - 1)
{
cout << a[q.front()] << ' ';
}
}
cout << endl;
for(int i = 0;i < n;i++)
{
while(!q2.empty() && a[q2.back()] < a[i])
{
q2.pop_back();
}
q2.push_back(i);
if(!q2.empty() && q2.front() < (i - k + 1))
{
q2.pop_front();
}
if(i >= k - 1)
{
cout << a[q2.front()] << ' ';
}
}
cout << endl;
return 0;
}
所以大家看懂了吗?
评论区回复