【leetcode】1052.爱生气的书店老板——滑动窗口

题目

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

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

题解&代码

//即求  不生气时间客户总数+其余时间constumer*grumpy之和  的最大值
//滑动窗口大小为x
//start——不生气开始时间，滑动窗口开始
//end——不生气结束时间，滑动窗口结尾

//在第 i 分钟生气，grumpy[i] = 1；不生气， grumpy[i] = 0
//为方便计算，Grumpy函数改为 生气grumpy[i]=0;不生气grumpy[i]=1
//则grumpy[i]*customers[i]表示第i分钟满意数

void Grumpy(int* grumpy,int grumpySize)
{
	for (int i = 0; i < grumpySize; i++)
	{
		grumpy[i] = 1 - grumpy[i];
	}
}


//不生气时间客户总数
int Sum(int* s, int start,int end)
{
	int sum = 0;
	for (int i = start; i <= end; i++)
	{
		sum += s[i];
	}
	return sum;
}

//其余时间满意数
int Sum2(int* customers, int* grumpy, int start, int end, int customersSize)
{
	int sum = 0;
	for (int i = 0; i < start; i++)
		sum += customers[i] * grumpy[i];
	for (int i = end+1; i < customersSize; i++)
		sum += customers[i] * grumpy[i];
	return sum;
}


int maxSatisfied(int* customers, int customersSize, int* grumpy, int grumpySize, int X) {
	Grumpy(grumpy, grumpySize);
	int start = 0;
	int end = X - 1;
	int sum = Sum(customers, start, end) + Sum2(customers, grumpy, start, end, customersSize);
	int maxSatisfied = sum;

	while (end < customersSize-1)
	{
		++end;
		sum = sum + customers[end] - customers[start] - customers[end] * grumpy[end] + customers[start] * grumpy[start];
		++start;
		if (maxSatisfied < sum)
			maxSatisfied = sum;
	}


	return maxSatisfied;
}