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

 

 

 

 

 

这个题还是一个滑动窗口的题：首先题目意思为

这个意思是：给出一个由正整数构成数组 A ，如果 A 中一个子数组恰好有 K 个不同的数字，那么就称之为好子数组。

该子数组可以有重复数字，比如当 K = 2 时，子数组[1,2,1]虽然有重复的 1，但是它总共有两个不同的数字，因此也

是个好子数组。题目就是让我们求一个数组A 中的好子数组的个数。（大概理解一下就是求最大值用滑动指针一个用来

记录最开始的位置，一个记录末尾位置）

可以这样理解;一开始left = right = 0，然后往右边滑动，并且里面的数据个数，判断是不是合格一直到不合格，然后left加一进行如上操作，最后

直到right = length(arr)退出循环，这一个整个流程就是求出K的max，然后再求出k-1的max然后进行减法运算即可

虽然是hard难度，但是和昨天那道题差不多，都是一个滑动窗口（双指针）

 

 

 

 

其实可以这样理解题意：恰好K个    <=====> 最多K个 - 最多K-1个

class Solution:


    def atMostWithKDistinct(self, A: List[int], K: int) -> int:

        n = len(A)                          # 数组长度
        freq = [0 for _ in range(n+1)]      # 每个数出现的次数
        count = 0                           # 出现不同类型数的个数

        res = 0
        left, right = 0, 0
        while right < n:
            if freq[A[right]] == 0:
                count += 1
            freq[A[right]] += 1
            right += 1
            while count > K:
                freq[A[left]] -= 1
                if freq[A[left]] == 0:
                    count -= 1
                left += 1
            # 最多 K 个
            res += right - left

        return res

    def subarraysWithKDistinct(self, A: List[int], K: int) -> int:
        return self.atMostWithKDistinct(A, K) - self.atMostWithKDistinct(A, K-1)