题目描述:
给定一个正整数数组 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
方法1:
主要思路:解题链接汇总
(1)参考官方题解,将题目转为求最多K个不同整数的子数组的个数;
class Solution {
public:
//最多K个不同整数的子数组
int find_most(vector<int>&A,int K){
unordered_map<int,int> mp;
int left=0,right=0;
int res=0;
while(right<A.size()){
++mp[A[right]];
++right;
if(mp.size()>K){
while(mp.size()!=K){
--mp[A[left]];
if(mp[A[left]]==0){
mp.erase(A[left]);
}
++left;
}
}
res+=right-left;
}
return res;
}
int subarraysWithKDistinct(vector<int>& A, int K) {
return find_most(A,K)-find_most(A,K-1);
}
};