题目描述
有一个 1×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≤k≤n≤10^3。
对于 50% 的数据,1≤k≤n≤10^4。
对于 100% 的数据,1≤k≤n≤2×10^6,矩阵中的元素大小不超过 10^4 并且均为正整数。
解释
一 判断队列尾项是否比a[i]大
- 是,将a[i]放进队列
- 否,去掉尾项
二 判断队列首项下标是否小于n-i+1
- 是,去掉首项
三 如果i>=m
- 是 输出首项
#include<bits/stdc++.h>
using namespace std;
const int N=2e6+10;
int n,m,a[N],f[N],head,tail=-1;
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=1;i<=n;i++){
while(head<=tail&&a[f[tail]]<a[i]) tail--;
f[++tail]=i;
while(head<=tail&&f[head]<i-m+1) head++;
if(i>=m) printf("%d\n",a[f[head]]);
}
return 0;
}
感谢浏览
/2024/3/4 发现代码错误,已更改。