LeetCode 992. K 个不同整数的子数组（双指针）

示例 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

class Solution {
public:
    int subarraysWithKDistinct(vector<int>& A, int K) {
        int l1 = 0, l2 = 0, r = 0, n = A.size();
        int ct1 = 0, ct2 = 0, ans = 0;
        vector<int> nums1(n+1, 0), nums2(n+1, 0);
        while(r < n)
        {
            if(++nums1[A[r]] == 1)
                ct1++;
            if(++nums2[A[r]] == 1)
                ct2++;
            while(ct1 > K)
            {
                if(--nums1[A[l1]] == 0)
                    ct1--;
                l1++;
            }
            while(ct2 >= K)
            {
                if(--nums2[A[l2]] == 0)
                    ct2--;
                l2++;
            }
            ans += l2-l1;
            r++;
        }
        return ans;
    }
};