（leetcode 992）k个不同整数的子数组

给定一个正整数数组 A，如果 A 的某个子数组中不同整数的个数恰好为 K，则称 A 的这个连续、不一定独立的子数组为好子数组。

（例如，[1,2,3,1,2] 中有 3 个不同的整数：1，2，以及 3。）

返回 A 中好子数组的数目。
输入：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].

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subarrays-with-k-different-integers
题解： 本题的意思是给定一个整数数组与一个k值，求其中满足不同整数个数为k的子数组的数量。如果我们采用暴力求解的话，最差的时间复杂度可能是0(n)*0(n)。为什么这么说呢？例如：当全部元素都相同，且k=1的时候，对于第一个元素开始的满足条件的子数组有n个，对于第二个元素开始的有n-1个…所以，最终的可得上面的时间复杂度。下面讲解双指针的解法： 我们以每个元素作为右边界进行子数组的筛选，这样有边界跑完整个数组以后，所得的答案就是我们需要的结果。对于每个右边界，其左边界并不是唯一的，可能存在多个左边界满足条件。那么，我们在右边界确定的情况下，如何求出满足的子数组的数量呢？我们可以这么做，即求出离右边界最远的左边界1与左边界2，他们之间的距离就是满足条件的个数，对于左边界1很好求，用left1表示。对于左边界2，实则上就是求满足k-1个不同整数的子数组的左边区间位置left2；
然后对于每个右边界，我们累加结果即可。代码如下：

class Solution {
    public int subarraysWithKDistinct(int[] A, int k) {
        int n = A.length;
        int left1 = 0, left2 = 0, right = 0, count1 = 0, count2 = 0, res = 0;
        int[] cur1 = new int[n+1];
        int[] cur2 = new int[n+1];
        while(right < n){
            if(cur1[A[right]] == 0) count1++;
            cur1[A[right]]++;
            if(cur2[A[right]] == 0) count2++;
            cur2[A[right]]++;
            while(count1 > k){
                if(--cur1[A[left1]] == 0) count1--;
                left1++;
            }
            while(count2 > k - 1){
                if(--cur2[A[left2]] == 0) count2--;
                left2++;
            }
            res += left2 - left1;
            right++;
        }
        return res;
    }
}