程序可在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
热门推荐
10 新年门口花束祝福语简短
11 盘锦结婚祝福语大全简短
12 父母生日祝福语 简短独特
13 家庭恩爱祝福语简短英文
14 高考俄语祝福语大全简短
15 虎年祝福语 诗句唯美简短
16 生日婚礼祝福语简短精辟
17 虎年喝酒拜年祝福语简短
18 教师闺蜜祝福语简短