leetcode_1052. Grumpy Bookstore Owner

- 日理万妓 2021-12-24 14:43 209阅读 0赞

1052. Grumpy Bookstore Owner

[https://leetcode.com/problems/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[0]==1)
                presum[0]=customers[0];
            for(int i=1;i<grumpy.size();i++)
            {
                presum[i]=grumpy[i]==1?(presum[i-1]+customers[i]):presum[i-1];
            }
            int max_sum=presum[X-1],last_pos=X-1;
            for(int i=X;i<grumpy.size();i++)
            {
                int st_pos = i-X+1;
                if(max_sum<presum[i]-presum[st_pos-1])
                {
                    max_sum=presum[i]-presum[st_pos-1];
                    last_pos=i;
                }
            }
            int res=0;
            for(int i=0;i<grumpy.size();i++)
            {
                if(grumpy[i]==0)
                    res+=customers[i];
                else if(i>=last_pos-X+1&&i<=last_pos)
                    res+=customers[i];
            }
            return res;
        }
    };

转载于:https://www.cnblogs.com/jasonlixuetao/p/10925898.html

[https_leetcode.com_problems_grumpy-bookstore-owner]: https://leetcode.com/problems/grumpy-bookstore-owner/