每个子数组的数字种类数

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

public class DistinctNumbersInSubarraysStream {
    // 方法: 使用Java Stream计算每个子数组的数字种类数
    public int[] distinctNumbers(int[] nums, int k) {
        // 用来存储结果的数组
        int[] result = new int[nums.length - k + 1];
        // 初始化窗口并计算初始种类数
        Set<Integer> set = Arrays.stream(nums, 0, k).boxed().collect(Collectors.toSet());
        result[0] = set.size();
        // 滑动窗口
        for (int i = k; i < nums.length; i++) {
            // 移除左边界的元素
            set.remove(nums[i - k]);
            // 添加右边界的元素
            set.add(nums[i]);
            // 记录结果
            result[i - k + 1] = set.size();
        }
        return result;
    }
}