用 Python 编写一个程序，通过 k 次操作查找包含重复数字的最长子列表的长度

pythonserver side programmingprogramming更新于 2025/12/18 10:20:17

假设我们有一个名为 nums 的列表和一个值 k，现在让我们考虑一个可以更新列表中任何数字的值的操作。我们必须执行最多 k 次操作后找到包含重复数字的最长子列表的长度。

因此，如果输入为 nums = [8, 6, 6, 4, 3, 6, 6] k = 2，则输出将为 6，因为我们可以将 4 和 3 更改为 6，以使该数组为 [8, 6, 6, 6, 6, 6, 6]，并且包含所有 6 的子列表的长度为 6。

要解决这个问题，我们将遵循以下步骤 −

如果 nums 为空，则
- 返回 0
num_count := 一个空映射
max_count := 0
start := 0
对于 nums 中的每个索引 end 和值 num，执行
- num_count[num] := num_count[num] + 1
- max_count := max_count 和 num_count[num] 中的最大值
- 如果 end - start + 1 > max_count + k，则
  - num_count[nums[start]] := num_count[nums[start]] - 1
  - start := start + 1
返回 end - start + 1

示例

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

from collections import defaultdict
def solve(nums, k):
   if not nums:
      return 0

   num_count = defaultdict(int)
   max_count = 0
   start = 0

   for end, num in enumerate(nums):
      num_count[num] += 1
      max_count = max(max_count, num_count[num])
      if end - start + 1 > max_count + k:
         num_count[nums[start]] -= 1
         start += 1
   return end - start + 1

nums = [8, 6, 6, 4, 3, 6, 6]
k = 2
print(solve(nums, k))