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

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

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

(例如,[1,2,3,1,2] 中有 3 个不同的整数:1,2,以及 3。)

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

解题:双指针

看到子数组,首选双指针,然后暴力解
(两个循环,判断计数)

class Solution:
    def subarraysWithKDistinct(self, A: List[int], K: int) -> int:
        left = right = tmp = res = 0
        sets = set()
        while left < len(A):
            while (len(sets)) <= K and right < len(A):
                sets.add(A[right])
                right += 1
                if len(sets) == K:
                    res += 1
            sets = set()
            left += 1
            right = left
        return res

但是这并没有通过测试案例,因超出了时间限制

然后我们发现这样的暴力解法确实太粗糙了

(1)首先我们每一次计数前都有一个判断,其实这个判断的条件很大程度上是涵盖在循环的条件里的,这意味着每个小循环都几乎多判断一次
(2)每次移动左指针时我们都重置了右指针即上一轮数据,但其实大可不必,因为K值固定,所以我们可以继承上一轮数据,优化算法

优化:滑动窗口

(这里未给出需要调用的库)

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

                while diff_nums > K:
                    res += n - j
                    counter[A[i]] -= 1
                    if counter[A[i]] == 0:
                        diff_nums -= 1
                    i += 1
            return res

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_51174011/article/details/113767177