Chương trình tìm số lượng danh sách con có tổng là mục tiêu đã cho trong python

Giả sử chúng ta có một danh sách các số được gọi là num và một mục tiêu giá trị khác, chúng ta phải tìm số danh sách con có tổng bằng với mục tiêu.

Vì vậy, nếu đầu vào là nums =[3, 0, 3] target =3, thì đầu ra sẽ là 4, vì chúng ta có các danh sách con này có tổng là 3:[3], [3, 0], [0, 3], [3].

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau:

• temp:=một bản đồ trống
• tạm thời [0]:=1
• s:=0
• ans:=0
• đối với tôi trong phạm vi từ 0 đến kích thước của nums, hãy thực hiện
  • s:=s + nums [i]
  • comp:=s - target
  • nếu comp là ở chế độ tạm thời, thì
    • ans:=ans + temp [comp]
  • temp [s]:=temp [s] + 1
• trả lại ans

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn:

Mã mẫu

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

Đầu vào

[3, 0, 3], 3

Đầu ra

4