力扣刷题笔记：992. K 个不同整数的子数组（滑窗法，很容易理解的代码、完整题解代码及注释）（暴力解法，结果python超时了。。。）

题目：

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

题解思路：

我们可以做出如下转化： 至少 K 个不同整数的子数组个数 - 至少 K+1 个不同整数的子数组个数

解决子问题：

（1）枚举 i，寻找以 i 为开头，至少 K 个不同整数的子数组个数
（2）如果我们发现了一个符合条件的 j，使得 [i,j] 是一个满足条件的子数组，那么 [i,j],[i,j+1],[i,j+2],…,[i,n-1] 都是满足条件的子数组。
（3）之后我们缩小窗口，直到窗口内的数组不满足条件了，我们再重新扩大窗口，回到上一步。

题解python代码：

class Solution:
    def subarraysWithKDistinct(self, A: List[int], K: int) -> int:
        def helper(A,K):
            n = len(A)
            i = 0
            res = 0
            ########################## 
            # 窗口内需要维护的变量
            diff_nums = 0
            counter = collections.defaultdict(int)
            ##########################
            for j in range(n):
                # 扩大窗口
                counter[A[j]] += 1
                if counter[A[j]] == 1:
                    diff_nums += 1

                # 如果找到以i开头满足条件的子数组了，就更新答案并缩小窗口
                while diff_nums > K:
                    res += n-j
                    counter[A[i]] -= 1
                    if counter[A[i]] == 0:
                        diff_nums -= 1
                    i += 1
            return res
        
        return -helper(A,K) + helper(A,K-1)

作者：MiloMusiala
链接：https://leetcode-cn.com/problems/subarrays-with-k-different-integers/solution/python-dong-hua-tong-xiang-shuang-zhi-zh-57ym/
来源：力扣（LeetCode）https://leetcode-cn.com/problems/subarrays-with-k-different-integers/

暴力解法思路：

获取A的所有子数组，在获取每一个子数组的同时判断其不同整数是否等于K，最后返回满足条件的子数组个数即可。

简单粗暴，可惜超时了。。。

实际做项目时可以用用，仅限参考

暴力解法python代码：

class Solution:
    def subarraysWithKDistinct(self, A: List[int], K: int) -> int:
        ans = 0
        n = len(A)
        # x + 1 表示子串的长度
        for x in range(n):
            # i 表示滑窗长度
            for i in range(x+1,n+1):
                # if len(set(A[x:i]))==K:
                #     ans += 1
                if len(set(A[x:i]))==K:
                    ans += 1
        return ans