题目:
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