目标和

一.题目描述

给定一个非负整数数组,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位整数。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/target-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

二.思路

每个数字前面有正负两种情况,总共2^n可能

使用动态规划,将当前状态分为两种情况:当前元素前面为正号或者负号,将当前需要满足的和作为当前状态

递推关系式:

DP[i][j]=DP[i-1][j-nums[i]]+DP[i-1][j+nums[i]]

DP[i][j]表示0-i个元素需要满足和为j的所有可能性

当nums[i]前面为负号时,当前情况的可能性跟0 to i-1的元素中满足和为j+nums[i]的情况数一样

当nums[i]为正号时也是一样的

需要注意第一个值为0 的情况,每个元素的需要求解的和的范围也是nums[0:i] to -nums[0:i]

代码如下:

class Solution:
    def findTargetSumWays(self, nums: List[int], S: int) -> int:
        Len=len(nums)
        Sum=sum(nums)
        if S>Sum:
            return 0
        Index1=[0 for i in range(S-Sum,S+Sum+1)]
        if nums[0]==0:
            Index1[nums[0]]=2
        else:
            Index1[-nums[0]]=1
            Index1[nums[0]]=1
        for i in range(1,Len):
            Index2=[0 for i in range(S-Sum,S+Sum+1)]
            for j in range(-sum(nums[:i+1]),sum(nums[:i+1])+1):
                Index2[j]=Index1[j-nums[i]]+Index1[j+nums[i]]
            Index1=copy.deepcopy(Index2)
        
        return Index1[S]

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

skj1995

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

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

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

打赏作者

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

抵扣说明:

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

余额充值