2100 适合打劫银行的日子

lsz冲呀

已于 2022-03-07 00:39:48 修改

阅读量53

点赞数

分类专栏： leetcode 文章标签： java leetcode

于 2022-03-07 00:36:27 首次发布

本文链接：https://blog.csdn.net/m0_61802230/article/details/123320788

版权

leetcode 专栏收录该内容

42 篇文章 1 订阅

订阅专栏

本文介绍了如何解决LeetCode上的一个算法问题——寻找适合打劫银行的日子。通过使用前缀和技巧，我们可以找到连续递增和递减的区间，然后判断是否满足打劫条件。代码中实现了遍历数组两次的方法，一次计算非递增前缀和非递减后缀，第二次遍历找出符合条件的日子。时间复杂度和空间复杂度均为线性。

摘要由CSDN通过智能技术生成

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 开始）。返回的日子可以任意顺序排列。

思路

本质上我们就是寻找连续递增递减的区间，很容易想到用前缀和计算。递减区间我们用类似于买卖股票的思路，可以通过一次遍历，即可得到。如果前面的数大于等于后面的数，区间长度就可以累加1；反之需要置0。递增类似，我们也可以倒序遍历。
我们将每个位置的对应的递增递减区间长度记录下来，再遍历一次判断每个位置且两边的递增递减长度大于time；即可得到每个“适合打劫银行”的日子

代码

public List<Integer> goodDaysToRobBank(int[] security, int time) {
        int n = security.length;
        // 非递增前缀
        int[] left = new int[n];
        // 非递减后缀
        int[] right = new int[n];
        // 一次遍历
        for (int i = 1; i < n; i++) {
            if (security[i] <= security[i - 1]) {
                left[i] = left[i - 1] + 1;
            }
            // 倒序计算
            if (security[n - i - 1] <= security[n - i]) {
                right[n - i - 1] = right[n - i] + 1;
            }
        }
       
        List<Integer> ans = new ArrayList<>();
        for (int i = time; i < n - time; i++) {
            // 判断左右递增递减数组
            if (left[i] >= time && right[i] >= time) {
                ans.add(i);    
            }
        }
        return ans;
    }

时间复杂度：O(n):只需遍历数组一遍
空间复杂度：O(n)：借助数组存储连续递增递减长度