Pythonで合計がターゲットに指定されているサブリストの数を見つけるプログラム

numsと呼ばれる数値のリストと別の値のターゲットがあるとすると、合計がターゲットと同じサブリストの数を見つける必要があります。

したがって、入力がnums =[3、0、3] target =3のようである場合、合計が3であるこれらのサブリストがあるため、出力は4になります。[3]、[3、0]、[0、 3]、[3]。

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

• temp：=空のマップ
• temp [0]：=1
• s：=0
• ans：=0
• 0からnumsのサイズの範囲のiの場合は、
  • s：=s + nums [i]
  • comp：=s-ターゲット
  • compが一時的なものである場合、
    • ans：=ans + temp [comp]
  • temp [s]：=temp [s] + 1
• 回答を返す

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

サンプルコード

from collections import defaultdict

class Solution:
   def solve(self, nums, target):
      temp = defaultdict(int)
      temp[0] = 1
      s = 0
      ans = 0
      for i in range(len(nums)):
         s += nums[i]
         comp = s - target
         if comp in temp:
            ans += temp[comp]
            temp[s] += 1
         return ans

ob = Solution()
nums = [3, 0, 3]
target = 3
print(ob.solve(nums, target))

入力

[3, 0, 3], 3

出力

4