给定一个正整数数组 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。
class Solution {
public int subarraysWithKDistinct(int[] A, int K) {
int l=0,r=0;
int n=A.length;
int ans=0,sum=0;
int[] nums=new int[n+1];
for(;r<n;r++) {
if(nums[A[r]]==0)
sum++;
nums[A[r]]++;
if(sum==K)
break;
}
if(r==n) return 0;
int num=1;
nums[A[r]]--;
if(nums[A[r]]==0) sum--;
for(;r<n;r++) {
if(nums[A[r]]==0) {
sum++;
num=1;
}
nums[A[r]]++;
while(sum>K) {
nums[A[l]]--;
if(nums[A[l]]==0)
sum--;
l++;
}
while(nums[A[l]]>1) {
nums[A[l]]--;
num++;
l++;
}
ans+=num;
}
return ans;
}
}