程序可在Python中精确计数具有k个唯一元素的子列表的数量

假设我们有一个称为nums的数字列表，另一个值为k，我们必须找到子列表的数量，要求该子列表中恰好有k个唯一数字。

因此，如果输入像nums = [2，2，3，4] k = 2，那么输出将是3，因为我们有如下子列表：[2，2，3]，[2，3]， [3，4]。

范例(Python)

让我们看下面的实现以更好地理解-

from collections import Counter
class Solution:
   def solve(self, nums, k):
      def count(K):
         slot = Counter()
         i = res = 0
         for j, x in enumerate(nums):
            slot[x] += 1
            while len(slot) > K:
               slot[nums[i]] -= 1
               if slot[nums[i]] == 0:
                  del slot[nums[i]]
               i += 1
            res += j - i + 1
         return res
      return count(k) - count(k - 1)
ob = Solution()
nums = [2, 2, 3, 4]
k = 2
print(ob.solve(nums, k))

输入值

[2, 2, 3, 4], 2

3