标签：integers Different LC int Subarrays exactly different

Given an array A of positive integers, call a (contiguous, not necessarily distinct) subarray of A good if the number of different integers in that subarray is exactly K.

(For example, [1,2,3,1,2] has 3 different integers: 1, 2, and 3.)

Return the number of good subarrays of A.

 

Example 1:

Input: A = [1,2,1,2,3], K = 2
Output: 7
Explanation: Subarrays formed with exactly 2 different integers: [1,2], [2,1], [1,2], [2,3], [1,2,1], [2,1,2], [1,2,1,2].

Example 2:

Input: A = [1,2,1,3,4], K = 3
Output: 3
Explanation: Subarrays formed with exactly 3 different integers: [1,2,1,3], [2,1,3], [1,3,4].

 

Note:

1. 1 <= A.length <= 20000
2. 1 <= A[i] <= A.length
3. 1 <= K <= A.length

 

 

Runtime: 128 ms, faster than 100.00% of C++ online submissions for Subarrays with K Different Integers. Memory Usage: 25 MB, less than 100.00% of C++ online submissions for Subarrays with K Different Integers.

 

class Solution {
public:
  int subarraysWithKDistinct(vector<int>& A, int K) {
    return atMostK(A, K) - atMostK(A,K-1);
  }
  int atMostK(vector<int>& A, int K) {
    int i = 0,res = 0;
    unordered_map<int,int> mp;
    for(int j=0; j<A.size(); j++) {
      if(!mp[A[j]]++) K--;
      while(K<0){
        if(!--mp[A[i]]) K++;
        i++;
      }   
      res += j - i + 1;
    }   
    return res;
  }
};

 

标签：integers,Different,LC,int,Subarrays,exactly,different
来源： https://www.cnblogs.com/ethanhong/p/10360555.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。