标签:hashmap 992 int res atMostKDistinct 个数 数组 滑动 Leetcode
直觉就是双指针算法,但我们发现,如果直接计算子数组中恰好有K个不同数的个数,是没有单调性的不能用双指针优化到O(N)
假设我们要求的不是恰好有K个,而是最多有K个不同的不同的数,那么就具有单调性。
把「恰好」改成「最多」就可以使用双指针一前一后交替向右的方法完成,这是因为 对于每一个确定的左边界,最多包含 K 种不同整数的右边界是唯一确定的,并且在左边界向右移动的过程中,右边界或者在原来的地方,或者在原来地方的右边。
因此我们实现一个辅助函数,计算最多包含K个数的子数组个数-最多包含k-1个子数组的个数就是最后答案。
对于给定区间[i,j], 固定住j,这个区间如果满足最多数的个数不超过K,那么所有满足要求的子数组个数是j-i+1。因此这样就可以算出的答案
class Solution {
public:
int subarraysWithKDistinct(vector<int>& A, int K) {
return atMostKDistinct(A,K)-atMostKDistinct(A,K-1);
}
int atMostKDistinct(vector<int>& A, int K){
unordered_map<int,int> hashmap;
int count = 0, res = 0;
for(int i=0,j=0;j<A.size();j++){
hashmap[A[j]]++;
if(hashmap[A[j]]==1) count++;
while(i<=j&&count>K){
hashmap[A[i]]--;
if(hashmap[A[i]]==0){
count--;
}
i++;
}
res+=(j-i+1);
}
return res;
}
};
标签:hashmap,992,int,res,atMostKDistinct,个数,数组,滑动,Leetcode 来源: https://blog.csdn.net/wwxy1995/article/details/113771190
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。