给定一个正整数数组 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
这个题还是一个滑动窗口的题:首先题目意思为
这个意思是:给出一个由正整数构成数组 A ,如果 A 中一个子数组恰好有 K 个不同的数字,那么就称之为好子数组。
该子数组可以有重复数字,比如当 K = 2 时,子数组[1,2,1]虽然有重复的 1,但是它总共有两个不同的数字,因此也
是个好子数组。题目就是让我们求一个数组A 中的好子数组的个数。(大概理解一下就是求最大值用滑动指针一个用来
记录最开始的位置,一个记录末尾位置)
可以这样理解;一开始left = right = 0,然后往右边滑动,并且里面的数据个数,判断是不是合格一直到不合格,然后left加一进行如上操作,最后
直到right = length(arr)退出循环,这一个整个流程就是求出K的max,然后再求出k-1的max然后进行减法运算即可
虽然是hard难度,但是和昨天那道题差不多,都是一个滑动窗口(双指针)
其实可以这样理解题意:恰好K个 <=====> 最多K个 - 最多K-1个
class Solution: def atMostWithKDistinct(self, A: List[int], K: int) -> int: n = len(A) # 数组长度 freq = [0 for _ in range(n+1)] # 每个数出现的次数 count = 0 # 出现不同类型数的个数 res = 0 left, right = 0, 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 # 最多 K 个 res += right - left return res def subarraysWithKDistinct(self, A: List[int], K: int) -> int: return self.atMostWithKDistinct(A, K) - self.atMostWithKDistinct(A, K-1)