LeetCode 每日一题992. K 个不同整数的子数组

992. K 个不同整数的子数组

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

方法一：滑动窗口

解题思路

老规矩，用双指针模拟「滑动窗口」。

这题换一个思路，求出「最多包含 K 个不同整数的子数组个数」和「最多包含 K - 1 个不同整数的子数组分数」，两者相减就是 「K 个不同整数的子数组个数」。

参考代码

public int subarraysWithKDistinct(int[] A, int K) {
    return atMostK(A, K) - atMostK(A, K - 1);
}


public int atMostK(int[] A, int K) {
    int n = A.length;
    int[] counts = new int[n + 1];

    int count = 0;
    int ret = 0;
    int left = 0, right = 0;
    while (right < n) {
        if (counts[A[right]] == 0) {
            count++;
        }
        counts[A[right]]++;
        right++;

        while (count > K) {
            counts[A[left]]--;
            if (counts[A[left]] == 0) {
                count--;
            }
            left++;
        }
        ret += right - left;
    }
    return ret;
}

执行结果