滑动窗口总结

滑动窗口

1.模型识别

给定一个大小为n的数组。有一个大小为 k 的滑动窗口，它从数组的最左边移动到最右边。你只能在窗口中看到 k 个数字。每次滑动窗口向右移动一个位置。

2.原理

维持一个单调队列(in most cases)。
不用每次移动都重新计算，而是要想办法利用重复信息，只对队首和队尾进行更新。

3.经典模板

1）求窗口最大最小值(LeetCode 239)

#include <iostream>
using namespace std;
const int N = 1000010;
int a[N], q[N];//q记录元素的下标

int main() {
    int n, k;
    scanf("%d%d", &n, &k);
    for (int i = 0; i < n; i ++ ) scanf("%d", &a[i]);

	//求最小值
    int hh = 0, tt = -1; //队首和队尾的指针
    //遍历数组元素
    for (int i = 0; i < n; i ++ ) {
    	//队列中有元素并且队列长度超过k时移动队首
        if (hh <= tt && i - k + 1 > q[hh]) hh ++ ;
		//更新队尾元素，删去比当前元素大的元素（保证队列单调递增，这样队首就是最小的元素）
        while (hh <= tt && a[q[tt]] >= a[i]) tt -- ;
        //插入新元素，移动队尾
        q[ ++ tt] = i;
		//移动到第k个元素就可以输出窗口最小值了
        if (i >= k - 1) printf("%d ", a[q[hh]]);
    }
    puts("");
    //求最大值
    hh = 0, tt = -1;
    for (int i = 0; i < n; i ++ ) {
        if (hh <= tt && 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]]);
    }
    puts("");
    return 0;
}

2）求窗口中位数(LeetCode 480)

• 思路：用两个有序容器来维护队列的前后两段，left保存小根堆，right保存大根堆，堆顶是分界点；插入时去判断插到哪一边，然后判断分界点的移动，保证两边元素个数最大差1.
• 代码逻辑：插入前k个数，用multiset作左右堆；然后滑动，插末尾，删开头，更新分界点

class Solution {
public:
    multiset<int> left, right;
    int k; //将窗口长度作为全局变量
    //计算中位数
    double getmedium() {
        if (k % 2 != 0) return *right.begin();
        else return ((double)*right.begin() + *left.rbegin()) / 2; //一个一个数转double，如果对加法结果转double会整数溢出
    }
    vector<double> medianSlidingWindow(vector<int>& nums, int _k) {
        k = _k;
        vector<double> res;
        //插入前k个数，先插到右边排好序，再把右边队头的一半移到左边
        for (int i = 0; i < k; i++) right.insert(nums[i]);
        for (int i = 0; i < k / 2; i++) {
            left.insert(*right.begin());
            right.erase(right.begin());
        }
        res.push_back(getmedium());
        //滑动窗口
        for (int i = k; i < nums.size(); i++) {
            int x = nums[i], y = nums[i - k]; //提取新加的元素和要删除的元素
            if (x >= *right.begin()) right.insert(x);
            else left.insert(x);
            if (y >= *right.begin()) right.erase(right.find(y)); //如果直接erase(y)会删除掉所有等于y的数，用find只删除一个等于y的数
            else left.erase(left.find(y));
            //更新两个堆，使两边个数相差不超过1
            while (left.size() > right.size()) {
                right.insert(*left.rbegin());
                left.erase(left.find(*left.rbegin()));//begin（）可以直接erase，rbegin（）不可以直接erase
            }
            while (right.size() > left.size() + 1) {
                left.insert(*right.begin());
                right.erase(right.begin());
            }
            res.push_back(getmedium());
        }
        return res;
    }
};

3）无重复字符的最长子串（LeetCode 3)

给定一个字符串，请你找出其中不含有重复字符的 最长子串 的长度。

• 思路：双指针，用一个哈希表维护当前窗口的不重复字母。每次窗口（左指针）向右移动一位，右指针移动到最远的无重复字母处。

class Solution {
public:
    int lengthOfLongestSubstring(string s) {
        int hh = 0, tt = 0;
        int res = 0;
        unordered_set<char> hash;
        while(tt < s.length()) {
        	//哈希表不空的时候，删除队头元素，移动左指针
            if(!hash.empty()) {
                hash.erase(s[hh]);
                hh++;
            }
            //队尾没有滑出并且最后一个元素不在窗口里的时候，添加该元素，移动右指针
            while(tt < s.length() &&  !hash.count(s[tt])) {
                hash.insert(s[tt]);
                tt++;
            }
            res = max(res, tt - hh);//每次记录当前不重复字符串的长度
        }
        return res;
    }
};