扫描
题目描述
有一个 1 × n 1 \times n 1×n 的矩阵,有 n n n 个整数。
现在给你一个可以盖住连续 k k k 个数的木板。
一开始木板盖住了矩阵的第 1 ∼ k 1 \sim k 1∼k 个数,每次将木板向右移动一个单位,直到右端与第 n n n 个数重合。
每次移动前输出被覆盖住的数字中最大的数是多少。
输入格式
第一行两个整数 n , k n,k n,k,表示共有 n n n 个数,木板可以盖住 k k k 个数。
第二行 n n n 个整数,表示矩阵中的元素。
输出格式
共 n − k + 1 n - k + 1 n−k+1 行,每行一个整数。
第 i i i 行表示第 i ∼ i + k − 1 i \sim i + k - 1 i∼i+k−1 个数中最大值是多少。
样例 #1
样例输入 #1
5 3
1 5 3 4 2
样例输出 #1
5
5
4
提示
对于 20 % 20\% 20% 的数据, 1 ≤ k ≤ n ≤ 1 0 3 1 \leq k \leq n \leq 10^3 1≤k≤n≤103。
对于 50 % 50\% 50% 的数据, 1 ≤ k ≤ n ≤ 1 0 4 1 \leq k \leq n \leq 10^4 1≤k≤n≤104。
对于 100 % 100\% 100% 的数据, 1 ≤ k ≤ n ≤ 2 × 1 0 6 1 \leq k \leq n \leq 2 \times 10^6 1≤k≤n≤2×106,矩阵中的元素大小不超过 1 0 4 10^4 104 并且均为正整数。
解题思路
这道题是典型的运用单调队列的算法,简单来说,单调队列就是在一个队列中的元素都是单调的,常用来解决增量区间 ( i − k , i ] (i-k, i] (i−k,i]内的最值问题。因此可以定义一个双向队列,所有在队尾的元素应是单调的,每向后压入一个元素时应保持单调性不变,若有不符合的,应当先弹出。对于队头元素,若已经不在这个区间内,则应将其弹出
代码
#include<iostream>
#include<deque>
using namespace std;
struct node
{
int x, id;
};
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
int n, k;
cin >> n >> k;
node num[n+1];
for(int i=1;i<=n;i++)
{
cin >> num[i].x;
num[i].id = i;
}
deque<node>qmax, qmin;
// 单调递减队列
for(int i=1;i<=n;i++)
{
while(!qmax.empty() && qmax.back().x < num[i].x)
{
qmax.pop_back();
}
qmax.push_back(num[i]);
if(qmax.front().id == i-k) qmax.pop_front();// 队头元素不在 (i-k, i]这个区间内
if(i>=k) cout << qmax.front().x<< endl;
}
// 单调递增序列
// for(int i=1;i<=n;i++)
// {
// while(!qmax.empty() && qmax.back().x > num[i].x)
// {
// qmax.pop_back();
// }
// qmax.push_back(num[i]);
// if(qmax.front().id == i-k) qmax.pop_front();// 队头元素不在 (i-k, i]这个区间内
// if(i>=k) cout << qmax.front().x<< endl;
// }
return 0;
}