leetcode算法题每日一练-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

分析

1. 根据题意可知，计算不同整数的子数组，很容易联想到滑动窗口。
2. 如示例1 [1,2,1,2,3]，设定左边界从数组下标0开始，题目知k为2，所以滑动窗口值为2.即可得，[1,2],[1,2,1],[1,2,1,2]。无法再继续下去，因为最后一个值为3已超出k=2的数量了。
3. 然后将左边界一刀下标1的位置，重复同上操作，可得[2,1],[2,1,2]
4. 继续移动边界为下标2即值为1，同上，得[1,2]
5. 同上，边界为下标3即值为2，得[2,3]
6. 左边界为最后数组最后一个值了，无法滑动了，所以终结计算返回结果。

实现

 public static int subarraysWithKDistinct(int[] A, int K) {
        // 记录窗口内不重复的整数
        List<Integer> contains = new ArrayList<>();
        // 计算最大子数组即返回结果
        int count = 0;
        for (int i = 0; i < A.length; i++) {
            // 清空数组，确保内循环开始是空数组
            contains.clear();
            for (int j = i; j < A.length; j++) {
                // 子数组内不重复数字的添加条件
                if ((contains.size() == 0 || contains.size() != K) && !contains.contains(A[j])) {
                    contains.add(A[j]);
                }
                // 滑动计算
                if (contains.contains(A[j]) && contains.size() == K) {
                    count++;
                }
                // 遇到边界终止循序如
                if (!contains.contains(A[j]) && contains.size() == K) {
                    break;
                }
            }
        }
        return count;
    }

结果

分析可得，两次遍历所以时间复杂度为O(N)的平方，空间复杂度用了一个list去存储，所以为O(N)。官方也提供了双指针滑动窗口计算方法，读者可以参考下。