LeetCode 6190. 找到所有好下标

6190. 找到所有好下标

 

【分析】分别用数组de和in记录以第i个数为结尾的最长非递增/递减的字数组的长度，然后从[k, n -k)枚举下标i，判断de[i - 1]和in[i + 1]是否>= k即可。

class Solution {

    // 前缀和
    // 3:40 24

    public List<Integer> goodIndices(int[] nums, int k) {
        int n = nums.length;
        int[] de = new int[n], in = new int[n];
        Arrays.fill(de, 1); Arrays.fill(in, 1);
        for (var i = 1; i < n; i++) {
            if (nums[i] <= nums[i - 1]) de[i] += de[i - 1];
        }
        for (var i = n - 2; i >= 0; i--) {
            if (nums[i] <= nums[i + 1]) in[i] += in[i + 1]; 
        }
        // 0 1 2 3 4 5 6
        // 2 1 1 1 3 4 1
        // 1 2 3 4 1 1 2
        // 1 2 2 2 2 1 1            
        List<Integer> ans = new ArrayList();
        for (var i = k; i < n - k; i++) {
            if (de[i - 1] >= k && in[i + 1] >= k) ans.add(i);
        }
        return ans;
    }
}

class Solution {

    // 前后缀分解
    // 4:58 3

    public List<Integer> goodIndices(int[] nums, int k) {
        int n = nums.length;
        int[] de = new int[n];
        int[] in = new int[n];
        de[0] = 1;
        for (var i = 1; i < n; i++) {
            de[i] = 1;
            if (nums[i] <= nums[i - 1]) de[i] += de[i - 1];
        }
        in[n - 1] = 1;
        for (var i = n - 2; i >= 0; i--) {
            in[i] = 1;
            if (nums[i] <= nums[i + 1]) in[i] += in[i + 1];
        }
        List<Integer> ans = new ArrayList();
        for (var i = k; i < n - k; i++) {
            if (de[i -1] >= k && in[i + 1] >= k) ans.add(i); 
        }
        return ans;
    }
}