LeetCode-2100. 适合打劫银行的日子

你和一群强盗准备打劫银行。给你一个下标从 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 开始）。返回的日子可以 任意 顺序排列。

思路：

• 首先处理两个特殊情况，当security没有time * 2 + 1多时，说明没有一天可以满足前后各time天，这时返回【】
• 如果time=0，返回security长度下表的列表
• 最后就是对security从【i-time:i+1】进行递减判断，对security从【i:i+time+1】进行递增判断，i是从time, len(security)-time进行循环，保证前后都有充足的time天

import numpy as np

class Solution:
    def goodDaysToRobBank(self, security: List[int], time: int) -> List[int]:
        if len(security) < (2 * time + 1):
            return []
        elif time == 0:
            return [i for i in range(len(security))]
        else:
            security_np = np.array(security)
            good_day = []
            for i in range(time, len(security)-time, 1):
                diff_pr = np.diff(security_np[i-time:i+1])
                diff_pa = np.diff(security_np[i:i+time+1])
                if (diff_pr <= 0).all() and (diff_pa >= 0).all():
                    good_day.append(i)
            return good_day

结果：

思路2：

• 1.利用前缀和的概念，也就是动态规划
• 2.如果当天比前一天小，那么当天left【i】位置等于left【i-1】+1；
• 3.如果当天比后一天小，那么当天right【n-1-i】位置等于left【n-i】+1；
• 4.最后判断当前位置i的left【i】和right【i】是否等于time就可以（有的方法，比如left【i】等于1或者不等于1，right【i】等于1或者不等于1来判断，等等这些判断方法，都和2 3 步骤有关系，但原理都一样）

class Solution:
    def goodDaysToRobBank(self, security: List[int], time: int) -> List[int]:
        n = len(security)
        left, right = [0] * n, [0] * n
        for i in range(1, n):
            if security[i - 1] >= security[i]:
                left[i] = left[i - 1] + 1
            if security[n - i] >= security[n - 1 - i]:
                right[n - 1 - i] = right[n - i] + 1
        return [i for i in range(n) if left[i] >= time and right[i] >= time]