Leetcode每日一题：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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路分析及代码实现

原问题可以转换成为求解「最多存在 K 个不同整数的子区间的个数」与 「最多存在 K - 1 个不同整数的子区间的个数」，他们两个的差值即是K个不同整数的子区间

class Solution:
    def subarraysWithKDistinct(self, A: List[int], K: int) -> int:
        return self.atMost(A, K) - self.atMost(A, K-1)
    
    def atMost(self, A, K):
        n = len(A) 
        freq = collections.Counter()
        left = 0
        right = 0
        count = 0
        res = 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
            res += right - left + 1
        return res

这道题自己没做出来，看了两遍官方题解才捋明白思路。。。。