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

 

关键字：滑动窗口，解析都在注释里面！

#include <iostream>
#include <vector>
using namespace std;


/* vector<int>: 生气：1， 不生气为：0 */

class Solution {
public:
	int maxSatisfied(vector<int>& customers, vector<int>& grumpy, int X) {
		int res = 0;

		if (grumpy.size() <= X) {
			for (int i = 0; i < customers.size(); i++) {
				res += customers[i];
			}
		}
		else {

			/* 滑动窗口,先算前X个为秘密技巧的窗口 */
			int i, j;
			int sum_tmp = 0;
			for (i = 0; i < X; i++) {
				sum_tmp += customers[i];
			}

			/* 计算为0，不生气的顾客 */
			for (j = i; j < grumpy.size(); j++) {
				if (grumpy[j] == 0) {
					sum_tmp += customers[j];
				}
			}

			/* 不要忽略第一个窗口。没错，我忽略了 */

			if (res < sum_tmp) {
				res = sum_tmp;
			}

			/* 维护长度为X的窗口：
			   grumpy[i] = 1，生气，说明之前在秘密技巧范围内，需要减去
			   grumpy[i+X] = 1，生气，现在要进入秘密技巧范围，所以加上
			   其他为0，不生气的顾客已经存过了。
			*/
			for (i = 0; i < (grumpy.size() - X); i++) {
				if (grumpy[i] == 1) {
					sum_tmp -= customers[i];
				}
				if (grumpy[i + X] == 1) {
					sum_tmp += customers[i + X];
				}

				if (res < sum_tmp) {
					res = sum_tmp;
				}
			}

			/* 暴力：超时 */
			/*
			for (int i = 0; i <= (grumpy.size() - X); i++) {
				int sum_tmp = 0;
				vector<int> gru_tmp(grumpy);

				for (int j = i; j < i + X; j++) {
					gru_tmp[j] = 0;
				}

				for (int j = 0; j < customers.size(); j++) {
					if (gru_tmp[j] == 0) {
						sum_tmp += customers[j];
					}
				}

				if (res < sum_tmp) {
					res = sum_tmp;
				}
			}

			*/

		}
		return res;
	}

};

int main() {

	vector<int> customers1 = { 1, 0, 1, 2, 1, 1, 7, 5 };
	vector<int> grumpy1 = { 0, 1, 0, 1, 0, 1, 0, 1 };
	int X1 = 3;

	vector<int> customers2 = { 4, 10, 10 };
	vector<int> grumpy2 = { 1, 1, 0 };
	int X2 = 2;

	Solution* ps = new Solution();

	cout << ps->maxSatisfied(customers2, grumpy2, X2) << endl;

	return 0;
}