题解 | #K 个不同整数子数组#

class Solution {
  private:
    // 函数用于计算具有最多 k 个不同整数的子数组数量
    int atMostKDistinct(const vector<int>& nums, int k) {
        unordered_map<int, int> count;
        int left = 0, result = 0;
        for (int right = 0; right < nums.size(); ++right) {
            if (count[nums[right]]++ == 0) k--;
            while (k < 0) {
                if (--count[nums[left]] == 0) k++;
                left++;
            }
            result += right - left + 1;
        }
        return result;
    }
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param nums int整型vector
     * @param k int整型
     * @return int整型
     */
    int nowsubarray(vector<int>& nums, int k) {
        // write code here
        return atMostKDistinct(nums, k) - atMostKDistinct(nums, k - 1);
    }
};