LeetCode 1052. 爱生气的书店老板(尺取)

题意:

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

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

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

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

数据范围:
1 <= X <= customers.length == grumpy.length <= 20000
0 <= customers[i] <= 1000
0 <= grumpy[i] <= 1

解法:

考虑X天最多能让多少个生气的客户变得不生气,
双指针维护一个大小为X的窗口,对窗口内生气客户的数量取max即可.
答案就是:max+不生气的数量.

code:

class Solution {
    
    
public:
    int maxSatisfied(vector<int>& c, vector<int>& g, int X) {
    
    
        int n=c.size();
        int tot=0;
        for(int i=0;i<n;i++){
    
    
            if(!g[i]){
    
    
                tot+=c[i];
            }
        }
        //
        int ans=0;
        int sum=0;
        for(int i=0;i<n;i++){
    
    
            if(g[i])sum+=c[i];
            if(i>=X&&g[i-X])sum-=c[i-X];
            ans=max(ans,sum);
        }
        return tot+ans;
    }
};

猜你喜欢

转载自blog.csdn.net/weixin_44178736/article/details/113968747