LeetCode 爱生气的老板

爱生气的老板

今天,书店老板有一家店打算试营业 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

这道题很明显就是用滑动窗口来进行解题。

下面就是实现代码

class Solution {
public:
    int maxSatisfied(vector<int>& customers, vector<int>& grumpy, int minutes) {
        //思路:滑动窗口
        int ans = 0;
        int size = customers.size();
        //首先将老板不生气时的顾客相加
        for(int i=0;i<size;i++){
            if(grumpy[i]==0){
                ans += customers[i];
            }
        }
        //然后进行滑动窗口部分
        //由于grumpy[i]==0不生气,grumpy[i]==1时生气,我们现在就是找生气数量最多的区间
        //下面问题转换成customer[i]*grumpy[i]
        //分两个部分进行相加,1. 当长度小于窗口长度,右指针直接相加。2. 当长度大于窗口长度,左指针右移。
        int best = 0;
        int cur = 0;
        for(int i=0,j=0;i<size;i++){
            cur += grumpy[i]*customers[i];
            if(i-j+1>minutes){
                cur -= customers[j]*grumpy[j];
                j++;
            }
            best = max(best,cur);
        }
        return best+ans;
    }
};
posted @ 2021-10-19 21:39  ChrisNg  阅读(39)  评论(0编辑  收藏  举报