leetcode 494. Target Sum(目标求和)

You are given a list of non-negative integers, a1, a2, …, an, and a target, S. Now you have 2 symbols + and -. For each integer, you should choose one from + and - as its new symbol.

Find out how many ways to assign symbols to make sum of integers equal to target S.

Example 1:

Input: nums is [1, 1, 1, 1, 1], S is 3.
Output: 5
Explanation:

-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

给出一个数组nums和一个目标和S,可以用+,-给数组中每个数字做运算符,问有多少种方法可以得到目标S

思路:
定义DP数组,dp[i][j]表示nums到下标i 为止的部分达到目标和j 有多少种方法。
定义nums的长度为n,那么结果只需要返回dp[n][S]

而所有可能的和的上限是数组nums中所有数字相加,下限是所有的数都取减号。
一个数字都不取的时候,即dp[0][-sum~sum] = 0
取一个数字的时候,dp[1][-nums[1]] = dp[1][nums[1]] = 1

同理取数字nums[i], 和为j时,上一步的和可以是j - nums[i], 也可以是j + nums[i], 因此dp[i][j] = dp[i-1][j - nums[i]] + dp[i-1][j + nums[i]]

因为数组下标不能为负值,因此负值的部分需要映射
即-sum~sum需要映射到0~2*sum
最后的和S也要映射到S+sum

    public int findTargetSumWays(int[] nums, int S) {
        if(nums == null || nums.length == 0) {
            return 0;
        }
        
        int sum = 0;
        for(int i = 0; i < nums.length; i++) {
            sum += nums[i];
        }
        
        int[][] dp = new int[nums.length + 1][2*sum + 1];
        
        for(int i = 1; i < nums.length; i++) {
            for(int j = 0; j < sum; j++) {               
                dp[i][j] = dp[i - 1][j - nums[i - 1] + sum] +
                    dp[i - 1][j + nums[i - 1] + sum];
            }
        }
        
        return dp[nums.length - 1][S + sum];
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

蓝羽飞鸟

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值