LeetCode——992. K 个不同整数的子数组(Subarrays with K Different Integers)——分析及代码（Java）

一、题目

给定一个正整数数组 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subarrays-with-k-different-integers
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、分析及代码

1. 滑动窗口

（1）思路

根据题目可发现好子数组的特点：当其右边界确定时，满足不同 K 个整数的左边界端点，处于一个连续的区间内。
因此，可设计两个拥有相同右边界的滑动窗口，使其区间内分别有 K 个和 K - 1 个不同的整数。当右边界每次移动时，两个窗口左边界位置差值即为当前右边界下好子数组的个数。

（2）代码

class Solution {
    public int subarraysWithKDistinct(int[] A, int K) {
        int n = A.length, ans = 0;
        //为滑动窗口设计2个左指针，指向不同整数个数为K和K-1的子数组左边界
        int[] num1 = new int[n + 1];
        int[] num2 = new int[n + 1];
        for (int i = 0; i <= n; i++) {
            num1[i] = 0;
            num2[i] = 0;
        }
        int l1 = 0, l2 = 0, r = 0;
        int diff1 = 0, diff2 = 0;
        while (r < n) {
            if (num1[A[r]]++ == 0)
                diff1++;
            if (num2[A[r++]]++ == 0)
                diff2++;
            while (diff1 > K)
                if (--num1[A[l1++]] == 0)
                    diff1--;
            while (diff2 > K - 1)
                if (--num2[A[l2++]] == 0)
                    diff2--;
            ans += l2 - l1;
        }
        return ans;
    }
}

（3）结果

执行用时 ：4 ms，在所有 Java 提交中击败了 97.66% 的用户；
内存消耗 ：41.8 MB，在所有 Java 提交中击败了 71.05% 的用户。

三、其他

暂无。