找出数组中的所有 K 近邻下标

// Java Stream Solution
// 思路：
// 1. 使用 IntStream 来遍历数组，找到所有等于 key 的下标。
// 2. 对于每个找到的下标，使用嵌套的 IntStream 标记该下标前后 k 距离内的下标为 K 近邻下标。
// 3. 使用 distinct 去重，然后排序并返回结果。

import java.util.*;
import java.util.stream.*;

public class Solution {
    public List<Integer> findKNearbyIndices(int[] nums, int key, int k) {
        return IntStream.range(0, nums.length)
            .filter(i -> nums[i] == key)
            .mapToObj(i -> IntStream.rangeClosed(Math.max(0, i - k), Math.min(nums.length - 1, i + k)))
            .flatMap(IntStream::boxed)
            .distinct()
            .sorted()
            .collect(Collectors.toList());
    }
}