992. K 个不同整数的子数组 滑动窗口

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

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

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

 

示例 1：

输入：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].
示例 2：

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

提示：

1 <= A.length <= 20000
1 <= A[i] <= A.length
1 <= K <= A.length

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subarrays-with-k-different-integers
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

/**
 * @param {number[]} A
 * @param {number} K
 * @return {number}
 */
var subarraysWithKDistinct = function(A, K) {
    // 最多k个连续 - 最多k-1连续
    function atMost(A, K) {
        let left = 0;
        let right = 0;
        let len = A.length;
        let freq = new Array(len + 1).fill(0);
        let count = 0;
        let res = 0
        // 左开右闭
        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++
            }
            res += right - left + 1 ;
        }
        return res
    }
    return atMost(A, K) - atMost(A, K - 1);
};