【前缀和】leetcode2100.适合打劫银行的日子

题目：
你和一群强盗准备打劫银行。给你一个下标从 0 开始的整数数组 security ，其中 security[i] 是第 i 天执勤警卫的数量。日子从 0 开始编号。同时给你一个整数 time 。

如果第 i 天满足以下所有条件，我们称它为一个适合打劫银行的日子：

• 第 i 天前和后都分别至少有 time 天。
• 第 i 天前连续 time 天警卫数目都是非递增的。
• 第 i 天后连续 time 天警卫数目都是非递减的。

更正式的，第 i 天是一个合适打劫银行的日子当且仅当：security[i - time] >= security[i - time + 1] >= … >= security[i] <= … <= security[i + time - 1] <= security[i + time].

请你返回一个数组，包含 所有 适合打劫银行的日子（下标从 0 开始）。返回的日子可以 任意 顺序排列。


解答：

class Solution:
    def goodDaysToRobBank(self, security: List[int], time: int) -> List[int]:
        res=[]
        n=len(security)
        if n<2*time+1:
            return res
        if time==0:
            return [i for i in range(n)]
        #使用两个数组来记录非递增和非递减的情况
        judge=[0]*n  
        for i in range(n-1):
            #-1：非递增
            if security[i]>security[i+1]:
                judge[i]=-1
            #1：非递减
            elif security[i]<security[i+1]:
                judge[i]=1
        #print(judge)
        #前缀和：使用a,b数组记录-1和1的数量
        a=[0]*(n+1)
        b=[0]*(n+1)
        for i in range(n):
            if judge[i]==-1:
                a[i+1]+=a[i]+1
            else:
                a[i+1]=a[i]
            if judge[i]==1:
                b[i+1]+=b[i]+1
            else:
                b[i+1]=b[i]
        #print(a)
        #print(b)
        #判断第time天是否能打劫
        for i in range(time,n-time):
            #判断[i-time:i]中1的数量为0
            #判断[i:i+time]中-1的数量为0
            if b[i]-b[i-time]==0 and a[i+time]-a[i]==0:
               res.append(i)            
        return res