初学python记录:力扣377. 组合总和 Ⅳ

题目:

给你一个由 不同 整数组成的数组 nums ,和一个目标整数 target 。请你从 nums 中找出并返回总和为 target 的元素组合的个数。

题目数据保证答案符合 32 位整数范围。

提示:

  • 1 <= nums.length <= 200
  • 1 <= nums[i] <= 1000
  • nums 中的所有元素 互不相同
  • 1 <= target <= 1000

思考:

又是前两天的题目的姊妹题,一开始试图用类似的方法写代码,如下:

python 复制代码
class Solution(object):
    def combinationSum4(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        n = len(nums)

        def makeTree(target, nums):
            if target < 0:
                return 0
            elif target == 0:
                return 1
            else:
                ans = 0
                for index in range(0, n):
                    ans += makeTree(target - nums[index], nums)
                return ans

        return makeTree(target, nums)

虽然是对的,但是超时了:

思考了一下,发现这道题可以用动态规划做。

优化:动态规划:

用dpi表示和为i的组合个数,i的范围是nums的最小值, target,最终dptarget即为题目所求,算法如下:

  1. 将nums数组从小到大排序,若 nums的最小值 > target,直接返回0;
  2. 将dp数组初始化为全0;
  3. 和为 nums的最小值(即nums0)的组合个数一定只有1个,所以dpnums\[0]初始化为1;
  4. 对每个i,遍历nums中的元素j,,此处注意判断i和j的大小关系(若j>i,当前dpi计算完毕,进行下一个i的计算;若j=i,说明有一个组合是只含一个元素j的,dpi加一;若j<i,则dpi += dpi-j
  5. 返回dptarget的值

代码如下:

python 复制代码
class Solution(object):
    def combinationSum4(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        n = len(nums)
        nums.sort()
        if target < nums[0]:
            return 0
        dp = [0 for _ in range(target+1)]
        dp[nums[0]] = 1
        for i in range(nums[0]+1, target+1):
            for j in nums:
                if j > i:
                    break
                elif j == i:
                    dp[i] += 1
                else:
                    dp[i] += dp[i-j]
        return dp[target]

提交通过:

相关推荐
投票竞赛1 分钟前
作品投票版权注意事项,线上评选上传图片视频须知
python·音视频
2601_962293535 分钟前
人工智能 & 神经网络完整入门路线(零基础可走,分阶段)
人工智能·python·深度学习·神经网络·机器学习
“AI国潮设计-小江”32 分钟前
【Python/SDXL实战】潮汕国潮IP视觉落地:普宁英歌舞猫IP & 创意甜品设计(附ComfyUI工作流与商业授权说明)
开发语言·人工智能·python·prompt·aigc
529宝宝起名网41 分钟前
用 Python 分析汉字字源与起名用字的关联规律:从六书结构到文化寓意的起名偏好洞察
开发语言·python
linx2951 小时前
单元九 · 零基础路线图-第 7–9 周·类与 RAII
c语言·开发语言·数据结构·c++·嵌入式硬件·算法
布莱克6051 小时前
C++ 七大排序算法详解:选择、冒泡、插入、计数、堆、快速与归并排序
数据结构·c++·算法·排序算法
一只旭宝1 小时前
Python 与 C/C++ 内存模型对比总结
c语言·c++·python
砚底藏山河1 小时前
容错重试与指数退避:网络抖动手抖不再丢数据(魔码量化实战 #04)
java·数据库·python·金融
shehuiyuelaiyuehao2 小时前
算法44,模拟算法,数青蛙
算法·哈希算法·散列表
今儿敲了吗2 小时前
02词云生成器
笔记·python