494. 目标和 golang 动态规划

题目

494. 目标和
给定一个非负整数数组，a1, a2, …, an, 和一个目标数，S。现在你有两个符号 + 和 -。对于数组中的任意一个整数，你都可以从 + 或 -中选择一个符号添加在前面。

返回可以使最终数组和为目标数 S 的所有添加符号的方法数。

示例 1:

输入: nums: [1, 1, 1, 1, 1], S: 3
输出: 5
解释:

-1+1+1+1+1 = 3
+1-1+1+1+1 = 3
+1+1-1+1+1 = 3
+1+1+1-1+1 = 3
+1+1+1+1-1 = 3

一共有5种方法让最终目标和为3。
注意:

数组非空，且长度不会超过20。
初始的数组的和不会超过1000。
保证返回的最终结果能被32位整数存下。

code

func findTargetSumWays(nums []int, S int) int {
	if len(nums)==0 || nums==nil{
		return 0
	}
	sum:=0
	for i:=0;i<len(nums);i++{
		sum+=nums[i]
	}


	// 1.恒不成立条件1
	if sum < S{
		return 0
	}


	// 2.恒不成立条件2

	sum+=S
	if sum%2!=0{
		return 0
	}


	dp:=make([]int,sum/2+1)
	dp[0]=1

	// 因此只要找到一个子集，令它们都取正号，并且和等于 (target + sum(nums))/2，就证明存在解
	for i:=0;i<len(nums);i++{
		for j:=sum/2;j>=nums[i];j--{
			dp[j]=dp[j]+dp[j-nums[i]]
		}
	}


	return dp[sum/2]
}