这是一个经典的单调序列的使用。单调队列,顾名思义是指队列内的元素是有序的,队头为当前的最大值(单调递减队列)或最小值(单调递增序列),以单调递减队列为例来看队列的入队和出队操作:
1、入队:
如果当前元素要进队,把当前元素和队尾元素比较,如果当前元素小于队尾元素,那么当前元素直接进队,如果当前元素大于队尾元素,那么队尾出队,将当前元素和新的队尾再做比较,直到当前元素大于队尾元素或者队列为空。单调队列只能在队尾插入元素,队尾和队头都可以删除元素。
2、出队:
出队直接取队头即可,因为用单调队列就是为了取最值,而队头就是最值。
因为每个元素都出队列进队列一次,而且进出没有多余的比较,所以平摊下来是O(n)的。
http://blog.csdn.net/f_x_q/article/details/10011805
实现采用了deque,就比较方便,也可以用数组记录头尾模拟。处理时注意结果元素的数组总共有n-k+1个就行了。
#include <cstdio> #include <queue> #define LEN 1000005 using namespace std; int num[LEN]; int max_ans[LEN]; int min_ans[LEN]; int n, k; void process_max() { deque<int> que; for (int i = 0; i < n; i++) { while (!que.empty() && que.front() <= i - k) { que.pop_front(); } while(!que.empty() && num[que.back()] < num[i]) { que.pop_back(); } que.push_back(i); if (i >= k -1) { max_ans[i - k + 1] = num[que.front()]; } } } void process_min() { deque<int> que; for (int i = 0; i < n; i++) { while (!que.empty() && que.front() <= i - k) { que.pop_front(); } while(!que.empty() && num[que.back()] > num[i]) { que.pop_back(); } que.push_back(i); if (i >= k -1) { min_ans[i - k + 1] = num[que.front()]; } } } int main() { scanf("%d", &n); scanf("%d", &k); for (int i = 0; i < n; i++) { scanf("%d", &num[i]); } process_max(); process_min(); for (int i = 0; i < n - k + 1; i++) { printf("%d ", min_ans[i]); } printf(" "); for (int i = 0; i < n - k + 1; i++) { printf("%d ", max_ans[i]); } printf(" "); }