力扣 2100. 适合打劫银行的日子

题目来源：https://leetcode-cn.com/problems/find-good-days-to-rob-the-bank/

大致题意：
给一个数组 arr 和整数 t，求出所有满足下列要求的元素的索引

• arr[i - t] >= arr[i -t + 1] >= … >= arr[i - 1] >= arr[i] <= arr[i + 1] <= …<= arr[i + t]

思路

双向遍历数组，正向求出截止目前索引的连续递减元素的数量，反向求出从当前索引开始的连续递增元素数量

然后遍历数组，若对于当前位置，前面的连续递减元素的数量大于等于 t，且后面连续递增元素数量大于等于 t，则当前位置满足题意

代码：

 public List<Integer> goodDaysToRobBank(int[] security, int time) {
        int n = security.length;
        int[] before = new int[n];
        int[] after = new int[n];
        before[0] = 0;
        after[n - 1] = 0;
        // 正向求出当前位置前面的连续递减元素的数量，且当前位置元素小于上一个位置
        for (int i = 1; i < n; i++) {
            if (security[i] <= security[i - 1]) {
                before[i] = before[i - 1] + 1;
            } else {
                before[i] = 0;
            }
        }
        // 反向求出当前位置后面的连续递增元素的数量，且当前位置元素小于下一个位置
        for (int i = n - 2; i >= 0; i--) {
            if (security[i] <= security[i + 1]) {
                after[i] = after[i + 1] + 1;
            } else {
                after[i] = 0;
            }
        }
        List<Integer> ans = new ArrayList<>();
        // 求出满足题意的位置
        for (int i = time; i < n - time; i++) {
            if (before[i] >= time && after[i] >= time) {
                ans.add(i);
            }
        }
        return ans;
    }