刷题-力扣-1052

1052. 爱生气的书店老板

题目链接

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/grumpy-bookstore-owner/
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题目描述

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

题目分析

  1. 根据题目描述,使用滑动窗口的思想
  2. 先求出所有老板不生气的满意量保存在total
  3. 在使用滑动窗口的思想,窗口大小为X,依次求出在窗口内的顾客满意量,最大存储在maxIncrease
  4. total+maxIncrease 即为所求

代码

class Solution {
public:
    int maxSatisfied(vector<int>& customers, vector<int>& grumpy, int X) {
        int total = 0;
        for (int i = 0; i < grumpy.size(); i++) {
            if (grumpy[i] == 0) total += customers[i];
        }
        int increase = 0;
        for (int i = 0; i < X; i++) {
            increase += customers[i] * grumpy[i];
        }
        int maxIncrease = increase;
        for (int i = X; i < grumpy.size(); i++) {
            increase = increase - customers[i - X] * grumpy[i - X] + customers[i] * grumpy[i];
            maxIncrease = maxIncrease > increase ? maxIncrease : increase;
        }
        return total + maxIncrease;
    }
};
上一篇:PostgreSQL查看系统视图的源码


下一篇:初识Vue3.0(3)——ref的使用