LeetCode 992 K个不同整数的子数组 HERODING的LeetCode之路

给定一个正整数数组 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路:
子数组恰好K个不同整数的子数组个数,可以理解为小于等于K个不同整数的子数组个数-小于等于K-1个不同整数的子数组个数,那么在编程时,只需要编写保证保存小于等于K个不同整数的子数组个数的函数,最后输入k与k-1,函数值做差即可,代码如下:

class Solution {
    
    
public:
    int GetMostDistinct(vector<int>& A, int K) {
    
    
        unordered_map<int, int> mp;
        int left = 0, right = 0, ret = 0;
        while (right < A.size()) {
    
    
            ++mp[A[right++]];
            while (mp.size() > K) {
    
    
                --mp[A[left]];
                if (mp[A[left]] == 0) mp.erase(A[left]);
                ++left;
            }
            // 如果这里改成 ret = max(ret, right - left),那么此函数就是 LeetCode 904 题的解。
            // 这里可以用动态规划的思想来理解 ret += right - left
            // right 增加一次,增加的数组个数可以用 right - left 来表示
            ret += right - left;
        }
        return ret;
    }

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

猜你喜欢

转载自blog.csdn.net/HERODING23/article/details/113765914