当前位置: 首页 > news >正文

动态规划 之 数组长度加长规避初始化

文章目录

在动态规划算法中,当我们的递推公式出现i-1,i-2的时候,一般的做法肯定是对于len(nums)<1,或者<2的时候特判一下
所以我们就采用将dp数组开长一点这样,就不用理初始化的问题,但是要根据题目的意思,对于dp数组本身的值进行赋值为0,还是一个很小的数
nums数组的下标是不用变化的

例子1:打家劫舍

198.打家劫舍
在这里插入图片描述

打家劫舍的递推公式是 dp[i] = max(dp[i-1]+nums[i],dp[i-2])
我们一般开的dp数组的长度是dp = [0]*len(nums),这样的话就会造成一些问题,就是当你的nums数组的长度小于3的时候,该公式使用不了,并且当你的数组的长度大于等于3的时候,前面的dp[0]和dp[1]的情况还需要特别判断

class Solution:
    def rob(self, nums: List[int]) -> int:
        # 可以使用记忆化搜索来做
        # dfs(i) = max(dfs(i-1),dfs(i-2)+nums[i])
        # 使用结果值作为返回值
        n = len(nums)
        dp = [0]*(n)
        if n <3:
            return max(nums)
        else:
            dp[0],dp[1] = nums[0],max(nums[0],nums[1])
            for i in range(2,n):
                dp[i] = max(dp[i-1],dp[i-2]+nums[i])
        return dp[n-1]

那么我们应该如何改进?
dp = [0]*(len(nums)+2),也就是说,将dp数组的长度开长,将对应的递推公式修改一下,也就是dp数组部分的下标都保证不会出现负数dp[i+2] = max(dp[i+1],dp[i]+nums[i])

class Solution:
    def rob(self, nums: List[int]) -> int:
        # 可以使用记忆化搜索来做
        # dfs(i) = max(dfs(i-1),dfs(i-2)+nums[i])
        # 使用结果值作为返回值
        n = len(nums)
        dp = [0]*(n+2)
        for i,c in enumerate(nums):
            # 原本是 dp[i] = max(dp[i-1],c+dp[i-2]),现在为了方便,dp方面每个都加上2
            dp[i+2] = max(dp[i+1],c+dp[i])
        return dp[-1]

例子2:最大子数组和

53.最大子数组和
在这里插入图片描述
在这里插入图片描述

这个递推公式是 dp[i] = max(dp[i-1]+nums[i],nums[i]),所以我们将dp数组开长一个,但是dp数组的初始值要赋值为一个很小的数,因为我们最后返回的是max(dp)

class Solution:
    def maxSubArray(self, nums: List[int]) -> int:
        # 定义dp[i]为以nums[i]结尾的最大连续子数组的和
        # dp[i] = max(dp[i-1]+nums[i],nums[i])
        n = len(nums)
        # 初始值设置为最小的值,这样后面返回max的时候才不会影响
        dp = [-10**4]*(n)
        dp[0] = nums[0]
        for i in range(1,n):
            dp[i] =  max(dp[i-1]+nums[i],nums[i])
        return max(dp)

相关文章:

  • internVL的本地部署微调实践——L2G4
  • 标准输入输出流,面向对象,构造函数
  • Kafka集群,常见MQ面试问题
  • 聚簇索引和非聚簇索引
  • 蓝桥杯 Java B 组之栈的应用(括号匹配、表达式求值)
  • Python 2 和 Python 3 在字符串编码上的差异
  • java八股文之Redis
  • 政务浏览器API文档及Demo大升级
  • uniapp二次封装组件(py组件)
  • SQL复习
  • 微调codegeex
  • 跟着AI学习vue3第二章
  • 机器视觉--switch语句
  • 基于N-gram模型的中文文本分析系统设计与实现
  • 什么是关系数据库理论?(函数依赖,范式)
  • 轻量级5G核心网:适应未来网络需求的关键方案
  • 【第11章:生成式AI与创意应用—11.2 音频与音乐生成的探索与实践】
  • C程序设计(第5版)——谭浩强(1)
  • lazarus 从资源文件中加载PNG图片
  • 【SpringBoot】深度解析 Spring Boot 拦截器:实现统一功能处理的关键路径
  • 澎湃研究所“营商环境研究伙伴计划”启动
  • 湖南省邵阳市副市长仇珂静主动向组织交代问题,接受审查调查
  • 趣看 | 五一黄金周:你拍风景,拍风景的人在拍你
  • 4月深圳新房、二手房成交同比均上涨,“5月有望延续积极向好的发展态势”
  • 全国铁路五一假期累计发送1.51亿人次,多项运输指标创历史新高
  • 两次蹚入同一条河,巴萨这一晚被命运抛弃