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 个不同整数的子数组分数」,两者相减就是 「K 个不同整数的子数组个数」。
参考代码
public int subarraysWithKDistinct(int[] A, int K) {
return atMostK(A, K) - atMostK(A, K - 1);
}
public int atMostK(int[] A, int K) {
int n = A.length;
int[] counts = new int[n + 1];
int count = 0;
int ret = 0;
int left = 0, right = 0;
while (right < n) {
if (counts[A[right]] == 0) {
count++;
}
counts[A[right]]++;
right++;
while (count > K) {
counts[A[left]]--;
if (counts[A[left]] == 0) {
count--;
}
left++;
}
ret += right - left;
}
return ret;
}
执行结果