LeetCode 992. Subarrays with K Different Integers（K 个不同整数的子数组） 滑动窗口/hard

---

1.Description

给定一个正整数数组 A，如果 A 的某个子数组中不同整数的个数恰好为 K，则称 A 的这个连续、不一定不同的子数组为好子数组。

（例如，[1,2,3,1,2] 中有 3 个不同的整数：1，2，以及 3。）

返回 A 中好子数组的数目。

---

2.Example

输入：A = [1,2,1,2,3], K = 2
输出：7
解释：恰好由 2 个不同整数组成的子数组：[1,2], [2,1], [1,2], [2,3], [1,2,1], [2,1,2], [1,2,1,2].

输入：A = [1,2,1,3,4], K = 3
输出：3
解释：恰好由 3 个不同整数组成的子数组：[1,2,1,3], [2,1,3], [1,3,4].

---

3.Solution

这一题主要难点在于想到：最多包含k个不同整数的子区间数-最多包含k-1个不同整数的子区间数=恰好包含k个不同整数的子区间数

因此这一题就是求恰好包含k个不同整数的子区间数。

对于代码中的res：为什么可以用新子数组的长度即 【right - left】来表示增加的子数组个数呢？
可以借鉴动态规划的思想，举个例子就好理解了：
当满足条件的子数组从 [A,B,C] 增加到 [A,B,C,D] 时，新子数组的长度为 4，同时增加的子数组为 [D], [C,D], [B,C,D], [A,B,C,D] 也为4。

public class Solution {

    public int subarraysWithKDistinct(int[] A, int K) {
        return atMostKDistinct(A, K) - atMostKDistinct(A, K - 1);
    }

    /**
     * @param A
     * @param K
     * @return 最多包含 K 个不同整数的子区间的个数
     */
    private int atMostKDistinct(int[] A, int K) {
        int len = A.length;
        int[] freq = new int[len + 1];//题中给的A的每一个元素的大小是0到A.length+1
        
        int left = 0;
        int right = 0;
        // [left, right) 里不同整数的个数
        int count = 0;
        int res = 0;
        // [left, right) 包含不同整数的个数小于等于 K
        while (right < len) {
            if (freq[A[right]] == 0) {
                count++;
            }
            freq[A[right]]++;
            right++;

            while (count > K) {
                freq[A[left]]--;
                if (freq[A[left]] == 0) {
                    count--;
                }
                left++;
            }
            // [left, right) 区间的长度就是对结果的贡献，就是从1到K个恰好的和就是最多K的值。
            res += right - left;//因为right在上面已经加一了，因此直接right-left就是左右边的距离了
            //这一步的原因见上边的题解。
        }
        return res;
    }
}