leetcode_1052. Grumpy Bookstore Owner

1052. Grumpy Bookstore Owner

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

题意：每个时刻i会有customer[i]个顾客进店，在i时刻店主的情绪是grumpy[i]，若grumpy[i]==1则店主脾气暴躁，那么顾客会不满意，否则顾客讲感到满意，店主会一个技巧保证连续X时间内不暴躁，但他只能使用一次。问最多可以有多少顾客满意。

---

解法：找到一个连续X时间段，使用技巧后，得到的收益最大。计算前缀和，然后依次遍历所有连续X时间段，找到使用技巧能使最多的顾客从不满意变为满意。

class Solution
{
public:
    int maxSatisfied(vector<int>& customers, vector<int>& grumpy, int X)
    {
        vector<int> presum(grumpy.size());
        if(grumpy[]==)
            presum[]=customers[];
        for(int i=;i<grumpy.size();i++)
        {
            presum[i]=grumpy[i]==?(presum[i-]+customers[i]):presum[i-];
        }
        int max_sum=presum[X-],last_pos=X-;
        for(int i=X;i<grumpy.size();i++)
        {
            int st_pos = i-X+;
            if(max_sum<presum[i]-presum[st_pos-])
            {
                max_sum=presum[i]-presum[st_pos-];
                last_pos=i;
            }
        }
        int res=;
        for(int i=;i<grumpy.size();i++)
        {
            if(grumpy[i]==)
                res+=customers[i];
            else if(i>=last_pos-X+&&i<=last_pos)
                res+=customers[i];
        }
        return res;
    }
};