LeetCode-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

📝思路

要说这题解是真的巧妙，精髓就在下图：“恰好”不好整，给整个“最多”，一下就变成老套路了——直接滑。

📝题解

int subarraysWithMostKDistinct(vector<int>& A, int K) {
    int len = A.size();
    int lft = 0, rgt = 0, res = 0, cnt = 0;
    vector<int> mark(len+1);

    while (rgt < len) {
        if (!mark[A[rgt]]) cnt++;
        mark[A[rgt]]++;
        while (cnt > K) {
            mark[A[lft]]--;
            if (!mark[A[lft]]) cnt--;
            ++lft;
        }
        res += rgt - lft;
        rgt++;
    }
    return res;
}

int subarraysWithKDistinct(vector<int>& A, int K) {
    return subarraysWithMostKDistinct(A, K) - subarraysWithMostKDistinct(A, K-1);
}