Java教程

算法基础课:滑动窗口

本文主要是介绍算法基础课:滑动窗口,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1e6 + 10;

int a[N], q[N];//q[N]维护的是数组下标。对应的a[q[i]]应当随着i的增加而增加/减少

int main()
{
    int n, k;
    cin >> n >> k;
    int hh = 0, tt = -1;
    for (int i = 0; i < n; i ++ ) 
    {
        scanf("%d", &a[i]);
        if (i - k + 1 > q[hh]) hh ++;//若队首不再被包含在窗口里
        while (hh <= tt && a[q[tt]] > a[i]) tt --;//当队尾(最大值)比下一个要读入的数还要大时
        q[++ tt] = i;//读入q中
        if (i >= k - 1) printf("%d ", a[q[hh]]);
    }
    cout << endl;
    hh = 0, tt = -1;
    for (int i = 0; i < n; i ++)
    {
        if (i - k + 1 > q[hh]) hh ++;
        while (hh <= tt && a[q[tt]] < a[i]) tt --;
        q[++ tt] = i;
        if (i >= k - 1) printf("%d ", a[q[hh]]);
    }
}
这篇关于算法基础课:滑动窗口的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!