您的位置:首页 > 其它

poj 2823(单调队列)

2015-08-13 17:25 232 查看
题意:有n个数字要按顺序进入一个窗口,窗口大小是k,问每次窗口内的k个数字的最小值和最大值。

题解:之前好像做过单调队列类型的题,但具体又忘了,所有再次学习一遍,这题应该是比较简单的模板题,维护队列内的值,保证输出都在队首,因为窗口大小是k,所以一旦队首的位置加k小于当前位置,就把队首向后移动。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1000005;
int n, k, a
, q
, pos
;

void get_min() {
int st = 1, en = 0;
for (int i = 0; i < k - 1; i++) {
while (st <= en && q[en] >= a[i])
en--;
q[++en] = a[i];
pos[en] = i;
}
for (int i = k - 1; i < n; i++) {
while (st <= en && q[en] >= a[i])
en--;
q[++en] = a[i];
pos[en] = i;
while (pos[st] + k < i + 1)
st++;
printf("%d", q[st]);
if (i < n - 1)
printf(" ");
else
printf("\n");
}
}

void get_max() {
int st = 1, en = 0;
for (int i = 0; i < k - 1; i++) {
while (st <= en && q[en] <= a[i])
en--;
q[++en] = a[i];
pos[en] = i;
}
for (int i = k - 1; i < n; i++) {
while (st <= en && q[en] <= a[i])
en--;
q[++en] = a[i];
pos[en] = i;
while (pos[st] + k < i + 1)
st++;
printf("%d", q[st]);
if (i < n - 1)
printf(" ");
else
printf("\n");
}
}

int main() {
while (scanf("%d%d", &n, &k) == 2) {
for (int i = 0; i < n; i++)
scanf("%d", &a[i]);
get_min();
get_max();
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: