结合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.

---

分析： 看到这道题，大致意思是有一个customers数组，每个数组值表示当天来书店的人数，还有一个grumpy数据表示老板在第i天是否生气（0表示不生气，1表示生气），还有个常数X表示老板最多能连续X天控制自己不生气，如果老板在第i天生气（即grumpy[i]==1）则当天顾客不满意，然后计算最多有多少客户能够感到满意的数量

分析转换：设计一个算法在适当的X天内控制不生气，即X区间内grumpy[x]内不管是0还是1都计为不生气，然后加上不在X区间内grumpy[i]==0的人数得出做大值。

---

当时看到道题，第一时间想到暴力解法，即从i=0开始只要grumpy[i]==1就将[i,i+x)(注意：i算是第一天，使用区间为左必右开)内人数相加，然后再加上[0，i-1]区间和[i+x,customers.length]区间内grumpy[i]==0的人数，大致算法如下：

class Solution {
  public int maxSatisfied(int[] customers, int[] grumpy, int X) {
      if(customers.length==1)
			return customers[0];
		int sum = -1;
		for(int i=0;i<grumpy.length;i++) {
			if(grumpy[i]==1) {
				int a = helper(customers,grumpy,i,i+X);  //计算X天区间内的人数
				System.out.println("a"+a);
				if(a>=sum)
					sum = a;    //取最多人数
			}
		}
      if(sum==-1) {            //表示grumpy数组中都是0，不存在1
			int num = 0;
			for(int i=0;i<customers.length;i++)
				num += customers[i];
			return num;
		}
		System.out.println(sum);
		return sum;
	}
	public int helper(int[] c,int[] g,int start,int end) {
		int s = 0;
		for(int i=0;i<start;i++){
			if(g[i]==0)
				s += c[i];
		}
		for(int i=start;i<g.length&&i<end;i++)
			s += c[i];
		for(int i=end;i<g.length;i++){
			if(g[i]==0)
				s += c[i];
		}
		return s;
	}
}

然后提交运行发现超出时间限制，自己算了下，平均时间复杂度确实是customers数组长度平方(n*n)，然后题目说明customsers数组长度最多为20000，确实算法比较耗时，后来类比求最长子数列和想到了滑动窗口算法，可以把平均复杂度降为n.

大致思路：

依照题意在[i,i+X)区间内，不管grumpy【i】是0或者1都计算到最多人数内，所以在此区间内若grumpy【i】==1也计算到最多人数，所以我们第一步可以把整个grumpy数组中grumpy【i】==0的人数先累加起来，因为不管有没有X区间或者X区间在哪里，grumpy【i】==0的人数这些都是需要加进去的，即：

for(int i=0;i<len;i++){
	if(grumpy[i]==0)                //累加整个数组中不生气的人数
		sum += customers[i];
}

然后（举例：输入：customers = [1,0,1,2,1,1,7,5], grumpy = [0,1,0,1,0,1,0,1], X = 3 输出：16）

先循环一次计算当grumpy[i]==0的人数，即为n=10；

从i==0开始，X==3：

1   0   1   2   1   1   7   5      cumstomers

0   1   0   1   0   1   0   1       grumpy

这时候定义一个变量max，max记录最初滑动窗口内grumpy[i]==1的人数，即为num = 0 所以max =n+num = 10 

接下来当i==3时，X===3进行遍历：(i为区间内最后位置)

1   0   1   2   1   1   7   5      cumstomers

0   1   0   1   0   1   0   1       grumpy

这时候定义一个变量n，n记录当前滑动窗口内grumpy[i]==1的人数， 并且在上次位置上向后移动了一位，如果grumpy[i-X]==1(即原来在窗口最前面的这次脱离了窗口，并且脱离力的grumpy[i-X]==1)，则num = num -cumstomers[i-X] 否则不处理

如果grumpy[i]==1(即这次新加入窗口的grumpy[i]==1)，则num =num +cumstomers[i],否则不处理,得到num =2;

max取n+num与上次的max之间的最大值，即max=Math.max(max,num+n)      max = 12

当i==4时，X===3进行遍历：(i为区间内最后位置)

1   0   1   2   1   1   7   5      cumstomers

0   1   0   1   0   1   0   1       grumpy

这时候定义一个变量n，n记录当前滑动窗口内grumpy[i]==1的人数， 并且在上次位置上向后移动了一位，如果grumpy[i-X]==1(即原来在窗口最前面的这次脱离了窗口，并且脱离力的grumpy[i-X]==1)，则num = num -cumstomers[i-X] 否则不处理

如果grumpy[i]==1(即这次新加入窗口的grumpy[i]==1)，则num =num +cumstomers[i],否则不处理,得到num =2;

max取n+num与上次的max之间的最大值，即max=Math.max(max,n+num)     max = 12

当i==5时，X===3进行遍历：(i为区间内最后位置)

1   0   1   2   1   1   7   5      cumstomers

0   1   0   1   0   1   0   1       grumpy

这时候定义一个变量n，n记录当前滑动窗口内grumpy[i]==1的人数， 并且在上次位置上向后移动了一位，如果grumpy[i-X]==1(即原来在窗口最前面的这次脱离了窗口，并且脱离力的grumpy[i-X]==1)，则num = num -cumstomers[i-X] 否则不处理

如果grumpy[i]==1(即这次新加入窗口的grumpy[i]==1)，则num =num +cumstomers[i],否则不处理,得到num =3;

max取n+num与上次的max之间的最大值，即max=Math.max(max,n+num)    max = 13

当i==6时，X===3进行遍历：(i为区间内最后位置)

1   0   1   2   1   1   7   5      cumstomers

0   1   0   1   0   1   0   1       grumpy

这时候定义一个变量n，n记录当前滑动窗口内grumpy[i]==1的人数， 并且在上次位置上向后移动了一位，如果grumpy[i-X]==1(即原来在窗口最前面的这次脱离了窗口，并且脱离力的grumpy[i-X]==1)，则num = num -cumstomers[i-X] 否则不处理

如果grumpy[i]==1(即这次新加入窗口的grumpy[i]==1)，则num =num +cumstomers[i],否则不处理,得到num =1

max取n+num与上次的max之间的最大值，即max=Math.max(max,n+num)    max = 13 

当i==7时，X===3进行遍历：(i为区间内最后位置)

1   0   1   2   1   1   7   5      cumstomers

0   1   0   1   0   1   0   1       grumpy

这时候定义一个变量n，n记录当前滑动窗口内grumpy[i]==1的人数， 并且在上次位置上向后移动了一位，如果grumpy[i-X]==1(即原来在窗口最前面的这次脱离了窗口，并且脱离力的grumpy[i-X]==1)，则num = num -cumstomers[i-X] 否则不处理

如果grumpy[i]==1(即这次新加入窗口的grumpy[i]==1)，则num =num +cumstomers[i],否则不处理,得到num =6;

max取n+num与上次的max之间的最大值，即  num = 6  max=Math.max(max,n+num)  max = 16

所以最终答案为16符合提交结果

下面为代码实现：

class Solution {
    public int maxSatisfied(int[] customers, int[] grumpy, int X) {
        
		int sum = 0;
		int len = customers.length;
		for(int i=0;i<len;i++){
			if(grumpy[i]==0)
				sum += customers[i];
		}
		int num = 0;
		for(int i=0;i<X;i++) {
			if(grumpy[i]==1) {
				num += customers[i];
			}
		}
		int max = sum+num;
		for(int i=X;i<len;i++) {
			if(grumpy[i-X]==1) 
				num -= customers[i-X];
			if(grumpy[i]==1)
				num += customers[i];
			max = Math.max(max, num+sum);
		}
		System.out.println(max);
		return max;
	
    }
}

：