力扣 1052. 爱生气的书店老板 思维 枚举 滑动窗口

https://leetcode-cn.com/problems/grumpy-bookstore-owner/

思路：硬要说的话感觉我这个解法怪怪的，和平时见到的滑动窗口不太一样，因为窗口大小是固定不变的。看完题目，很容易想到使用一个长度固定为$X$的窗口，然后枚举左边界即可，那么怎么保证整体复杂度是$O(n)$的呢？利用前缀和的思想，在初始化时以当前窗口的左右边界(初始时窗口为$[0,X)$)为限，可以计算出一个前缀和和后缀和，那么在移动窗口的时候维护这两个和值即可。

class Solution {
public:
    int maxSatisfied(vector<int>& customers, vector<int>& grumpy, int X) {
        int prefix=0,suffix=0,middle=0,n=customers.size();
        for(int i=0;i<X;i++)
            middle+=customers[i];
        for(int i=X;i<n;i++)
            if(!grumpy[i])
                suffix+=customers[i];
        int ans=middle+suffix;
        for(int i=X;i<n;i++)
        {
            if(grumpy[i])
                middle+=customers[i];
            if(grumpy[i-X])
                middle-=customers[i-X];
            ans=max(ans,prefix+middle+suffix);
        }
        return ans;
    }
};