程序可在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