2020-12-11

题目描述:给定一个非负整数数组,a1, a2, …, an, 和一个目标数,S。现在你有两个符号 + 和 -。对于数组中的任意一个整数,你都可以从 + 或 -中选择一个符号添加在前面。
返回可以使最终数组和为目标数 S 的所有添加符号的方法数。
解题思路一:动态规划,遍历所有可以取到的值,通过对当前值进行加和减进行转移,代码如下:

class Solution:
    def findTargetSumWays(self, nums: List[int], S: int) -> int:
        n = len(nums)
        total = sum(nums)
        dp = [[0]*(2*total+1) for _ in range(n+1)]
        dp[0][total] = 1
        for i in range(1, n+1):
            for j in range(2*total+1):
                if j- nums[i-1] >= 0:
                    dp[i][j] += dp[i-1][j-nums[i-1]]
                if j+nums[i-1] <= 2 * total:
                    dp[i][j] += dp[i-1][j+nums[i-1]]
        return dp[-1][S+total] if total >= S else 0

解题思路二:枚举所有可能情况,结果等于S就+1,用递归进行遍历,python会时间超时,代码如下:

class Solution:
    count = 0
    def calculate(self, nums, i, total, S):
        if i == len(nums):
            if total == S:
                self.count += 1
        else:
            self.calculate(nums, i+1, total+nums[i], S)
            self.calculate(nums, i+1, total-nums[i], S)
    def findTargetSumWays(self, nums: List[int], S: int) -> int:
        # count = 0
        
        self.calculate(nums, 0, 0, S)
        return self.count

解题思路三:转化为背包问题,把所有符号为正的数总和设为一个背包的容量,容量为x;把所有符号为负的数总和设为一个背包的容量,容量为y。在给定的数组中,有多少种选择方法让背包装满。令sum为数组的总和,则x+y = sum。而两个背包的差为S,则x-y=S。从而求得x=(S+sum)/2。基于上述分析,题目转换为背包问题:给定一个数组和一个容量为x的背包,求有多少种方式让背包装满。代码如下:

class Solution:
    def findTargetSumWays(self, nums: List[int], S: int) -> int:
        total = sum(nums)
        if total  < S or (S + total) % 2: return 0
        target = (S + total) // 2
        dp = [0] * (target + 1)
        dp[0] = 1
        for num in nums:
            for i in range(target, num-1, -1):
                dp[i] += dp[i-num]
        return dp[-1]
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值