Pythonのバイナリリストで合計kのサブリストの数を見つけるプログラム

0または1のバイナリリストがあるとします。 kという別の入力もあります。合計がkと同じサブリストの数を見つける必要があります。

したがって、入力がnums =[1、0、0、1、1、1、0、1] k =3の場合、サブリストは[1,0,0,1,1]であるため、出力は8になります。 ]、[0,0,1,1,1]、[0,0,1,1,1,0]、[0,1,1,1]、[0,1,1,1,0]、 [1,1,1]、[1,1,1,0][1,1,0,1]。

これを解決するには、次の手順に従います-

• sums：=マップには、最初はキー0のvalye1が含まれています
• r_sum：=0
• ans：=0
• numsのxごとに、
  • r_sum：=r_sum + x
  • ans：=ans +（（r_sum --k）が存在する場合はsums [r_sum --k]、それ以外の場合は0）
  • sums [r_sum]：=1 +（（r_sum --k）が存在する場合はsums [r_sum --k]、それ以外の場合は0）
• 回答を返す

例

理解を深めるために、次の実装を見てみましょう-

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))

入力

[1, 0, 0, 1, 1, 1, 0, 1], 3

出力

8