题目描述
有一个 1 ∗ n 的矩阵,有 n 个正整数。
现在给你一个可以盖住连续的 k 的数的木板。
一开始木板盖住了矩阵的第 1 ∼ k 个数,每次将木板向右移动一个单位,直到右端与第 n 个数重合。
每次移动前输出被覆盖住的最大的数是多少。
输入格式
第一行两个数,n,k,表示共有 n 个数,木板可以盖住 k 个数。
第二行 n 个数,表示矩阵中的元素。
输出格式
共 n − k + 1 行,每行一个正整数。
第 i 行表示第 i ∼ i + k − 1 个数中最大值是多少。
输入输出样例
输入 #1
5 3 1 5 3 4 2
输出 #1
5 5 4
说明/提示
对于 20% 的数据保证:1 ≤ n ≤ 1e3,1 ≤ k ≤ n
对于 50% 的数据保证:1 ≤ n ≤ 1e4,1 ≤ k ≤ n
对于 100% 的数据保证:1 ≤ n ≤ 2 ∗ 1e6,1 ≤ k ≤ n
矩阵中元素大小不超过 1e4。
思路
单调队列。
单调队列是一种特殊的队列,它的元素不超过m个,且元素在其中单调递增或递减。
比如这道题求最大值,在插入某个元素之前,先检查队列尾部的元素是否比它小,如果小,那么把队尾删除。 再检查队头,是否还在那个区间内。如果在区间外部,删除。 最后才是插入这个元素。
按这种算法,就可以维护队列的单调性。
之前在P1440用过单调队列,那个是求连续区间最小值。这里是求最大值。
还有一道经典的题目:P1886 滑动窗口,也是单调队列实现的。
其实这些算法都可以用树状数组,线段树,ST表啥的去搞。但是那么一大坨代码,肯定没有单调队列实现的快啦!不过单调队列虽然代码少,还是很难理解的。
学会了一种新方法实现单调队列:双向队列(deque)
#include <iostream>
#include <stdio.h>
#include <deque>
using namespace std;
int n,m;
struct node
{
int val;//存储序列中该元素的值
int position;//位置
}a[2000001];
deque<node> dq;//声明双向队列
int maxn[2000001];//最大值序列
int main()
{
//freopen("testdata.in","r",stdin);
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
register int i,j;
cin>>n>>m;
for(i=1;i<=n;i++)
{
cin>>a[i].val;
a[i].position=i;
}
for(i=1;i<=n;i++)//从位置开始检索,从i=2开始也可以过
{
while(!dq.empty() && dq.back().val<a[i].val)//队列不为空,且如果队尾小于a[i]
dq.pop_back();//就不要队尾了,pop掉
dq.push_back(a[i]);//把a[i]放进去
while(!dq.empty() && dq.front().position<=i-m)//检查队头是否在区间外
dq.pop_front();//是就pop
maxn[i]=dq.front().val;//记录max
}
for(i=1;i<=n;i++)
{
if(i>=m)//如果是个完整的区间
{
cout<<maxn[i]<<endl;
}
}
}