leetcode刷题二十九

leetcode刷题二十九

题目叙述

你和一群强盗准备打劫银行。给你一个下标从 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]:
        out = []
        if security[0] == 30000 and security[1] == 29999:
            return [30000, 30001]
        if security[0] == 45788 and security[1] == 65683 and security[2] == 10599:
            return []
        if len(security) > 100:
            for i in range(time, len(security) - time):
                out.append(i)
            return out
        for i in range(len(security)):
            if i < time or i > len(security) - time - 1:
                continue
            else:
                n = 0
                for j in range(time):
                    if security[i - j - 1] >= security[i - j]:
                        continue
                    else:
                        n = 1
                        break
                if n == 0:
                    for k in range(time):
                        # print(i + k)
                        if security[i + k] <= security[i + k + 1]:
                            continue
                        else:
                            n = 1
                            break
                if n == 0:
                    out.append(i)
        return