Leetcode————992.K个不同整数的子数组

题目

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

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

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

解题思路

维护一个滑动窗口，用hash表记录窗口中的元素个数

代码

class Solution:
    def subarraysWithKDistinct(self, A: List[int], K: int) -> int:
        max_num = max(A)
        count = [0] * (max_num + 1)
        start,cur,ans,res,n = 0,0,0,0,1
        for end in range(len(A)):
            if not count[A[end]]:
                if cur < K:
                    count[A[end]] = 1
                    cur += 1
                else:
                    while start < end and count[A[start]]:
                        count[A[start]] -= 1
                        start += 1
                        if not count[A[start - 1]]: break
                    count[A[end]] = 1
                    n = 1
            else:
                count[A[end]] += 1
            if cur == K:
                while start < end and count[A[start]] > 1:
                    count[A[start]] -= 1
                    start += 1
                    n += 1
                res += n
        return res