992. K 个不同整数的子数组（问题转化 + 双指针）

本文链接：https://blog.csdn.net/qq_43765535/article/details/113768486

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

在这里插入图片描述

双指针，但是不能直接用

题目要求不同整数个数恰好为 k 个，这样双指针做不到。

将问题转化成最多 k 个，这样最多 k 个不同整数的个数就分解成恰好1个 + 恰好 2 个 + 恰好 …

所以就能变成

在这里插入图片描述

AC Code

class Solution {
    public int subarraysWithKDistinct(int[] a, int k) {
        return atmostk(a, k) - atmostk(a, k - 1);
    }

    /**
     * 最多 k 个不同整数的子数组个数
     */
    public int atmostk(int[] a, int k){
        int len = a.length;
        // 1 <= A[i] <= A.length
        // 记录频数
        int[] map = new int[len + 1];
        int cnt = 0, ans = 0;
        int left = 0, right = 0;

        while(right < len) {
            if(map[a[right]] == 0) cnt++;
            map[a[right]]++;
            right++;

            while(cnt > k) {
                map[a[left]]--;
                if(map[a[left]] == 0) cnt--;
                left++;
            }

            // right - left 表示增加的子数组
            // 当满足条件的子数组从 [A,B,C] 增加到 [A,B,C,D] 时，新子数组的长度为 4
            // 同时增加的子数组为 [D], [C,D], [B,C,D], [A,B,C,D] 也为4。
            //  这些在之前的j已经加上啦
            ans += (right - left);
        }

        return ans;
    }

}