1052. 爱生气的书店老板

一.题目描述

今天，书店老板有一家店打算试营业 customers.length 分钟。每分钟都有一些顾客（customers[i]）会进入书店，所有这些顾客都会在那一分钟结束后离开。

在某些时候，书店老板会生气。 如果书店老板在第 i 分钟生气，那么 grumpy[i] = 1，否则 grumpy[i] = 0。 当书店老板生气时，那一分钟的顾客就会不满意，不生气则他们是满意的。

书店老板知道一个秘密技巧，能抑制自己的情绪，可以让自己连续 X 分钟不生气，但却只能使用一次。

请你返回这一天营业下来，最多有多少客户能够感到满意的数量。

示例：
输入：customers = [1,0,1,2,1,1,7,5], grumpy = [0,1,0,1,0,1,0,1], X = 3
输出：16
解释：
书店老板在最后 3 分钟保持冷静。
感到满意的最大客户数量 = 1 + 1 + 1 + 1 + 7 + 5 = 16.

提示：
1 <= X <= customers.length == grumpy.length <= 20000
0 <= customers[i] <= 1000
0 <= grumpy[i] <= 1

来源：力扣（LeetCode）
链接：题目来源

二.问题分析

1）这个题目看起来有点长，但是仔细读题后就会发现表达的意思其实非常简单。就是有两个相同长度的数组，数组grumpy中所有元素均为0和1，为0时customers数组对应位置的元素有效，反之为1则无效。然后现在有一个长度X，可以修改grumpy中连续的X个数，将其中的1变成0，我们要求解所有为0的位置对应的元素总和。
2）这里比较巧妙的一个地方就是，可以用(1-grumpy[i])*customers[i]来表示对应grumpy[i]位置的元素是否被需要。
3）因为有连续X个grumpy位置的值可以发生改变，所以我们使用滑动窗口来处理问题。

三.代码

class Solution {
public:
    int maxSatisfied(vector<int>& customers, vector<int>& grumpy, int X) {
    	//res为初始状态下对应满意的客户，ext为使用技巧后增加的满意客户量
    	int res=0,ext=0,ans=0;
        //l为窗口左端，r为窗口右端
        int l=0,r=0;
        while(r<customers.size())
        {
            //初始情况下满意的客户量
            res += (1-grumpy[r])*customers[r];
            //滑动窗口
            ans += grumpy[r]*customers[r++];
            ext = max(ext,ans); //找可以修改的最大值
            if(r-l==X) ans-= grumpy[l]*customers[l++]; //这时窗口向右移，减去最左边的那个数。
        }
        return res+ext;
    }
};