找到所有好下标

/*
 * 思路：
 * 1. 使用两个数组 left 和 right 分别记录每个位置之前和之后的非递增和非递减长度。
 * 2. 遍历数组，根据条件找到所有的好下标。
 */

import java.util.ArrayList;
import java.util.List;
import java.util.stream.IntStream;

public class GoodIndicesStream {
    public List<Integer> goodIndices(int[] nums, int k) {
        int n = nums.length;
        int[] left = new int[n];
        int[] right = new int[n];
        List<Integer> result = new ArrayList<>();
        
        // 初始化left数组
        IntStream.range(0, n).forEach(i -> {
            if (i == 0 || nums[i] <= nums[i - 1]) {
                left[i] = (i == 0) ? 1 : left[i - 1] + 1;
            } else {
                left[i] = 1;
            }
        });
        
        // 初始化right数组
        IntStream.iterate(n - 1, i -> i >= 0, i -> i - 1).forEach(i -> {
            if (i == n - 1 || nums[i] <= nums[i + 1]) {
                right[i] = (i == n - 1) ? 1 : right[i + 1] + 1;
            } else {
                right[i] = 1;
            }
        });
        
        // 查找好下标
        IntStream.range(k, n - k).filter(i -> left[i - 1] >= k && right[i + 1] >= k).forEach(result::add);
        
        return result;
    }
    
    public static void main(String[] args) {
        GoodIndicesStream solution = new GoodIndicesStream();
        int[] nums = {2, 1, 1, 1, 3, 4, 1};
        int k = 2;
        System.out.println(solution.goodIndices(nums, k)); // 输出: [2, 3]
    }
}