Leetcode 2100.适合劫银行的日子

最新推荐文章于 2025-12-03 23:35:37 发布

原创最新推荐文章于 2025-12-03 23:35:37 发布 · 7k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #职场和发展

算法专栏收录该内容

16 篇文章

订阅专栏

这篇博客探讨了一种解决LeetCode上的'找到好日子抢劫银行'问题的方法。通过计算数组的前缀和，确定在给定的时间范围内，安全值下降和上升的次数，以找出合适的抢劫日。代码中使用了动态规划和前缀和技巧，避免了不安全的抢劫时段。

链接： https://leetcode-cn.com/problems/find-good-days-to-rob-the-bank/

思路：

前缀和：
    g数组：表示下标i与i - 1 的大小关系
    规定：security[i] > security[i - 1] g[i] = 1;
        security[i] < security[i - 1] g[i] = -1;
        security[i] = security[i - 1] g[i] = 0;
    在 [i - time, i + time) 内满足：
        (i - time, i] 1的个数为0
        (i, i + time] -1的个数为0
    需要左开右闭区间，使用前缀和时，用i + 1位置记录i的状态；a[i]表示i 与 i - 1关系，但是此时i 和 i - 1的关系并不关心

代码：

   public List<Integer> goodDaysToRobBank(int[] security, int time) {
        if (security.length < 2 * time + 1) return new ArrayList<>();
        List<Integer> res = new ArrayList<>();

        int len = security.length;
        int a[] = new int[len + 1];
        int b[] = new int[len + 1];
        int g[] = new int[len];
        for (int i = 1; i < len; i++) {
            if (security[i] == security[i - 1]) continue;
            g[i] = security[i] < security[i - 1] ? -1 : 1;
        }

        for (int i = 1; i <= len ;i++) a[i] = a[i - 1] + (g[i - 1] == 1 ? 1 : 0);
        for (int i = 1; i <= len ;i++) b[i] = b[i - 1] + (g[i - 1] == -1 ? 1 : 0);

        for (int i = time; i < len - time; i++) {
            if (a[i + 1] - a[i - time + 1] > 0) continue;
            if (b[i + 1 + time] - b[i + 1] > 0) continue;
            res.add(i);
        }
        return res;
    }