【题目描述】:
给定一个长度为n的数列a,再给定一个长度为k的滑动窗口,从第一个数字开始依次框定k个数字,求每次框定的数字中的最大值和最小值,依次输出所有的这些值。
【输入描述】:
第一行包含两个整数 n 和 k ,分别表示数组的长度和滑动窗口长度。
第二行n个整数,表示数列元素的值。
【输出描述】:
第一行从左到右窗口看到的最小值。
第二行从左到右窗口看到的最大值。
【样例输入】:
8 3
1 3 -1 -3 5 3 6 7
【样例输出】:
-1 -3 -3 -3 3 3
3 3 5 5 6 7
【时间限制、数据范围及描述】:
时间:1s 空间:64M
30%:n<=100 k<=20
60%:n<=5000 k<=20
100%:n<=10^6,每个元素不操过int类型
需要读入输出优化
【AC代码】:
#include<bits/stdc++.h>
#define M(a,b) memset(a,b,sizeof(a))
#define INF 0x3f3f3f3f
#define Mod 19650827
using namespace std;
inline void read(int &x){
char ch=getchar(),c=ch;
x=0;
while(ch<'0' || ch>'9'){
c=ch;
ch=getchar();
}
while(ch>='0' && ch<='9'){
x=(x<<1)+(x<<3)+ch-'0';
ch=getchar();
}
if(c=='-')x=-x;
}
const int maxn = 1e6 + 10;
inline void write(int x) {
if(x<0) putchar('-'),x=-x;
if(x>9) write(x/10);
putchar(x%10+'0');
}
struct node {
int pos, value;
node() {}
node(int pos,int value){
this->pos = pos;
this->value = value;
}
} p[maxn];
int n, m;
int a[maxn];
void get_max() {
int head = 0, tail = 0;
for (int i = 1; i <= n; ++i) {
while(head < tail && i - p[head].pos >= m) head++;
while(head < tail && a[i] > p[tail - 1].value) tail--;
p[tail++] = node(i,a[i]);
if (i >= m) {
write(p[head].value);
putchar(i == n?'\n':' ');
}
}
}
void get_min() {
int head = 0, tail = 0;
for (int i = 1; i <= n; ++i) {
while(head < tail && i - p[head].pos >= m) head++;
while(head < tail && a[i] < p[tail - 1].value) tail--;
p[tail++] = node(i, a[i]);
if (i >= m) {
write(p[head].value);
putchar(i == n?'\n':' ');
}
}
}
int main() {
read(n),read(m);
for(int i = 1; i <= n; ++i)read(a[i]);
get_min();
get_max();
return 0;
}