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