Giả sử chúng ta có một danh sách các số không âm được gọi là num và cũng có một mục tiêu là số nguyên. Chúng ta phải tìm số cách sắp xếp + và - trong các số sao cho biểu thức bằng với đích.
Vì vậy, nếu đầu vào giống như nums =[2, 3, 3, 3, 2] target =9, thì đầu ra sẽ là 2, vì chúng ta có thể có -2 + 3 + 3 + 3 + 2 và 2 + 3 + 3 + 3 - 2.
Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau:
-
s:=tổng của tất cả các số trong nums
-
nếu (s + target) mod 2 không giống 0 hoặc target> s thì
-
trả về 0
-
-
W:=thương số của (s + target) / 2
-
dp1:=danh sách kích thước (W + 1) và điền bằng 0
-
dp1 [0]:=1
-
dp2:=Một danh sách kích thước (W + 1) và điền bằng 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
-
đối với j trong phạm vi 0 đến W + 1, thực hiện
-
nếu j> =nums [i], thì
-
dp2 [j]:=dp2 [j] + dp1 [j - nums [i]]
-
-
-
đối với j trong phạm vi 0 đến W + 1, thực hiện
-
dp1 [j]:=dp1 [j] + dp2 [j]
-
dp2 [j]:=0
-
-
-
trả về phần tử cuối cùng của dp1
Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn:
Ví dụ
class Solution: def solve(self, nums, target): s = sum(nums) if (s + target) % 2 != 0 or target > s: return 0 W = (s + target) // 2 dp1 = [0] * (W + 1) dp1[0] = 1 dp2 = [0] * (W + 1) for i in range(len(nums)): for j in range(W + 1): if j >= nums[i]: dp2[j] += dp1[j - nums[i]] for j in range(W + 1): dp1[j] += dp2[j] dp2[j] = 0 return dp1[-1] ob = Solution() nums = [2, 3, 3, 3, 2] target = 9 print(ob.solve(nums, target))
Đầu vào
[2, 3, 3, 3, 2], 9
Đầu ra
2