K 个不同整数的子数组

/*
 * 思路：
 * 我们使用滑动窗口来找到包含恰好k个不同整数的子数组。
 * 在每一步中，我们将窗口右端扩展并记录每个整数的出现次数。
 * 如果窗口内的不同整数个数大于k，则移动左端以减少不同整数的个数。
 * 最终，当窗口内的不同整数个数正好为k时，记录子数组的数量。
 */
import java.util.*;
import java.util.stream.*;

public class GoodSubarraysStream {
    public long countGoodSubarrays(int[] nums, int k) {
        Map<Integer, Integer> countMap = new HashMap<>();
        int left = 0;
        long result = 0;
        for (int right = 0; right < nums.length; right++) {
            countMap.put(nums[right], countMap.getOrDefault(nums[right], 0) + 1);
            while (countMap.size() > k) {
                countMap.put(nums[left], countMap.get(nums[left]) - 1);
                if (countMap.get(nums[left]) == 0) {
                    countMap.remove(nums[left]);
                }
                left++;
            }
            if (countMap.size() == k) {
                result += right - left + 1;
            }
        }
        return result;
    }

    public static void main(String[] args) {
        GoodSubarraysStream gss = new GoodSubarraysStream();
        System.out.println(gss.countGoodSubarrays(new int[]{1, 2, 1, 2, 3}, 2)); // 输出7
        System.out.println(gss.countGoodSubarrays(new int[]{1, 2, 1, 3, 4}, 3)); // 输出3
    }
}