力扣992.K个不同整数的子数组——Python

本个题目我们采用双指针的方法进行解决， 但是这里有需要注意的地方，根据 好子数组 的定义，当我们使用双指针的方法时，会发现当固定左边端点时，符合 恰好有 KK 个不同整数 的子区间的右端点并不是唯一的。

这里我们转换下思路，假设求 最多有 K 个不同整数 的子区间，最终求得符合要求的子区间中，包含以下情况：

仅包含 1 个不同整数的子区间；
仅包含 2 个不同整数的子区间；
…
仅包含 K-1 个不同整数的子区间；
仅包含 K 个不同整数的子区间。
我们可以发现 恰好有 K 个不同整数 的子区间，包含于求 最多有 KK 个不同整数 的子区间的情况中。如果将前面 K−1 种情况剔除，也就剩下 恰好有 KK 个不同整数 的子区间这一种情况。

而我们可以发现前面 K−1 情况，其实就是求 最多有K−1 个不同整数 的子区间。

那么最终要求的：【恰好有 K 个不同整数 的子区间个数】 = 【最多有 K 个不同整数 的子区间个数】- 【最多有K−1 个不同整数 的子区间个数】。

		def at_most_K_distinct(A, K):
            """求最多 K 个不同整数子区间的个数
            """
            n = len(A)

            left = 0
            right = 0
            # 存储元素出现次数
            freq = collections.defaultdict(int)
            # 记录不同整数的个数
            count = 0
            ans = 0
            while right < n:
                # 当元素第一次出现时，记录不同整数的个数
                if freq[A[right]] == 0:
                    count += 1
                # 更新整数出现的次数
                freq[A[right]] += 1
                # 移动 right 指针
                right += 1
                # 当不同整数个数大于 K 时，移动左指针 left
                # 注意更新整数出现的次数
                while count > K:
                    freq[A[left]] -= 1
                    # 当数字出现次数为 0 时，表示区间内少了一个不同整数
                    # 令 count - 1
                    if freq[A[left]] == 0:
                        count -= 1
                    left += 1
                # 这里的累加，是因为移动指针，符合要求的右端点被纳入进来后，新增的子数组个数恰好等于 right - left
                # (注意 right 在前面先进行移动，所以此时 right 在新纳入的右端点的右侧)
                ans += right - left
            
            return ans
        
        return at_most_K_distinct(A, K) - at_most_K_distinct(A, K-1)