前缀和 LeetCode2100. 适合打劫银行的日子

最新推荐文章于 2024-10-27 05:36:49 发布

18阿鲁

最新推荐文章于 2024-10-27 05:36:49 发布

阅读量159

点赞数

本文链接：https://blog.csdn.net/weixin_43260719/article/details/125922458

版权

动态规划前缀和银行抢劫连续天数安全策略

关键词由CSDN通过智能技术生成

2100. 适合打劫银行的日子

分析

前缀和

class Solution {
    public List<Integer> goodDaysToRobBank(int[] security, int time) {
int n = security.length;
        int[] dp1 = new int[n];
        int[] dp2 = new int[n];
        dp1[0] = 0;
        dp2[n-1] = 0;
        for (int i = 1; i < n; i++) {
            if (security[i] <= security[i-1]) {
                dp1[i] = dp1[i-1] + 1;
            }
        }
        for (int i = n -2; i >= 0; i--) {
            if (security[i] <= security[i+1]) {
                dp2[i] = dp2[i+1] + 1; 
            }
        }
        List<Integer> list = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            if (dp1[i] >= time && dp2[i] >= time) {
                list.add(i);
            }
        }
        int[] ans = new int[list.size()];
        for (int i = 0; i < list.size(); i++) {
            ans[i] = list.get(i);
        }
        return list;
    }
}