Java 8在一段时间内的滑动窗口计数

import java.util.HashMap;
import java.util.Map;

public class SlidingWindowCount {
    public static void main(String[] args) {
        int[] nums = {1, 2, 1, 3, 4, 2, 3};
        int k = 3; // 窗口大小

        Map<Integer, Integer> countMap = new HashMap<>(); // 用于存储元素计数的Map
        int distinctCount = 0; // 不同元素的计数

        // 初始化窗口
        for (int i = 0; i < k; i++) {
            int num = nums[i];
            countMap.put(num, countMap.getOrDefault(num, 0) + 1);
            if (countMap.get(num) == 1) {
                distinctCount++;
            }
        }

        System.out.println("窗口内的不同元素数量：" + distinctCount);

        // 移动窗口
        for (int i = k; i < nums.length; i++) {
            int numToRemove = nums[i - k];
            int numToAdd = nums[i];

            countMap.put(numToRemove, countMap.get(numToRemove) - 1);
            if (countMap.get(numToRemove) == 0) {
                distinctCount--;
            }

            countMap.put(numToAdd, countMap.getOrDefault(numToAdd, 0) + 1);
            if (countMap.get(numToAdd) == 1) {
                distinctCount++;
            }

            System.out.println("窗口内的不同元素数量：" + distinctCount);
        }
    }
}