用 Python 编写程序，在二进制列表中查找总和为 k 的子列表数量

pythonserver side programmingprogramming更新于 2026/1/11 8:12:17

假设我们有一个包含 0 或 1 的二进制列表。我们还有另一个输入称为 k，我们必须找到总和与 k 相同的子列表的数量。

因此，如果输入为 nums = [1, 0, 0, 1, 1, 1, 0, 1] k = 3，则输出将为 8，因为子列表为 [1,0,0,1,1], [0,0,1,1,1], [0,0,1,1,1,0], [0,1,1,1,1], [0,1,1,1,0], [1,1,1], [1,1,1,0] [1,1,0,1]。

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

sums := a map 最初包含键 0 的 valye 1
r_sum := 0
ans := 0
对于 nums 中的每个 x，执行
- r_sum := r_sum + x
- ans := ans + (sums[r_sum - k] 如果 (r_sum - k) 存在，则为 0)
- sums[r_sum] := 1 + (sums[r_sum - k] 如果 (r_sum - k) 存在，则为 0)
返回 ans

示例

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

def solve(nums, k):
   sums = {0: 1}
   r_sum = 0
   ans = 0

   for x in nums:
      r_sum += x
      ans += sums.get(r_sum - k, 0)
      sums[r_sum] = sums.get(r_sum, 0) + 1

   return ans

nums = [1, 0, 0, 1, 1, 1, 0, 1]
k = 3
print(solve(nums, k))