码迷,mamicode.com
首页 > 其他好文 > 详细

关于动态规划的问题494_LEETCODE_TARGET_SUM

时间:2018-10-09 15:04:24      阅读:144      评论:0      收藏:0      [点我收藏+]

标签:return   ==   ret   思考   elf   ndt   leetcode   时间复杂度   动态规划   

在做这道题的时候,思考了很久一直不知道怎么做,如果全部遍历的话肯定会出现TLE超时问题,为什么呢?

1、每个值都可以是取或者不取,那么就有2^n^组合方法,则时间复杂度O(2^n^),随着数量的增长,时间成指数级增长

2、所以放弃了遍历想法

 

于是乎考虑了动态规划

动态规划:
1、起始和的字典 dp[0] = {0:1}
2、第1个数+-1,此时和的字典为 dp[1] = {1:1,-1:1}
3、第2个数+-1,此时和的字典为 dp[2] = {2:1,0:2,-2:1}
4、第3个数+-1,此时和的字典为 dp[3] = {3:1,1:3,-1:3,-3:1},target为1时,那么返回dp[3][1],即3
比如dp[3][1] = dp[2][2]+dp[2][0] = 1 + 2 = 3
和字典的动态转义方程 dp[i][j] = dp[i-1][j-num] + dp[i-1][j+num]
class Solution:
    def findTargetSumWays(self, nums, S):
        """
        :type nums: List[int]
        :type S: int
        :rtype: int
        """
        # 动态规划:
        # 1、起始和的字典             dp[0] = {0:1}
        # 2、第1个数+-1,此时和的字典为 dp[1] = {1:1,-1:1}
        # 3、第2个数+-1,此时和的字典为 dp[2] = {2:1,0:2,-2:1}
        # 4、第3个数+-1,此时和的字典为 dp[3] = {3:1,1:3,-1:3,-3:1},target为1时,那么返回dp[3][1],即3
        # 比如dp[3][1] = dp[2][2]+dp[2][0] = 1 + 2 = 3
        # 和字典的动态转义方程  dp[i][j] = dp[i-1][j-num] + dp[i-1][j+num]
        sums = 0
        for n in nums:
            sums += n
        sum_diff = sums - S
        if sum_diff < 0 or sum_diff % 2 != 0:
            return 0

        dp = {0: 1}
        for n in nums:
            temp = {}
            for sign in (1, -1):
                for j in dp:
                    key = j + n * sign
                    if key not in temp:
                        value = dp[j]
                    else:
                        value = temp[key] + dp[j]
                    temp[key] = value
            dp = temp
        return dp[S]


if __name__ == __main__:
    s = Solution()
    a = s.findTargetSumWays([1,1,1,1,1], 3)
    print(a)

 




关于动态规划的问题494_LEETCODE_TARGET_SUM

标签:return   ==   ret   思考   elf   ndt   leetcode   时间复杂度   动态规划   

原文地址:https://www.cnblogs.com/ceshixuexi/p/9760156.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!