动态规划,英文:Dynamic Programming,简称DP,如果某一问题有很多重叠子问题,使用动态规划是最有效的。
动态规划中每一个状态一定是由上一个状态推导出来的。而贪心算法不同,贪心没有状态推导,而是从局部直接选最优的。
动态规划问题,将被拆解为如下五步:
注意: 有些题目非常相似,但实际上做法不同,比如1.5.4.2 里的 leetcode的 5. 最长回文子串与牛客网的HJ32 密码截取都是求最长回文的子串,它是字符串,而 1.5.4.3里的 516. 最长回文子序列求的是子序列的长度,它是数字。所以,求长度则使用动态规划,求字符串则使用回溯,贪心等等其他算法,当然使用动态规划也是可以做出来的,但是时间复杂度相对高一些。
力扣题目链接
F(0) = 0,F(1) = 1
F(n) = F(n - 1) + F(n - 2),其中 n > 1
1一般解法:递归:
class Solution:
def fib(self, n: int) -> int:
if n < 2:
return n
else:
return self.fib(n-1) + self.fib(n-2)
时间复杂度:O(2^n)
空间复杂度:O(n)
2高级解法:动态规划:
0 1 1 2 3 5 8 13 21 34 55
如果代码写出来,发现结果不对,就把dp数组打印出来看看和我们推导的数列是不是一致的。
综上,代码如下:
class Solution:
def fib(self, n: int) -> int:
if n < 2:
return n
dp = [0]*(n+1)
dp[0] = 0
dp[1] = 1
for i in range(2, n+1):
dp[i] = dp[i-1]+dp[i-2]
return dp[n]
时间复杂度:O(n)
空间复杂度:O(n)
上面我们维护了一个长度为N+1的数组,实际上我们只需要维护两个数值就行,因为我们只需要最后一个数。修改数组为两个数值:
class Solution:
def fib(self, n: int) -> int:
if n < 2:
return n
dp1, p2 = 0, 1
for i in range(2, n+1):
dp1, dp2 = dp2, dp1+dp2
return dp2
时间复杂度:O(n)
空间复杂度:O(1)
力扣题目链接
1一般解法:递归(超时):
class Solution:
def climbStairs(self, n: int) -> int:
if n<2:
return n
else:
return self.climbStairs(n-1) + self.climbStairs(n-2)
2高级解法:动态规划:
根据规则代码如下:
class Solution:
def climbStairs(self, n: int) -> int:
# n为1 和 2
if n<3:
return n
# 初始化
# 一阶楼梯
dp1 = 1
# 二阶楼梯
dp2 = 2
# 递归公式
for i in range(3, n+1):
dp1, dp2 = dp2, dp1+dp2
return dp2
力扣题目链接
高级解法:动态规划:
dp0, dp1 = cost[0], cost[1]
确定遍历顺序
因为是模拟台阶,而且dp[i]又dp[i-1]dp[i-2]推出,所以是从前到后遍历cost数组就可以了。
举例推导dp数组
拿示例2:cost = [1, 100, 1, 1, 1, 100, 1, 1, 100, 1] ,来模拟一下dp数组的状态变化,如下:
class Solution:
def minCostClimbingStairs(self, cost: List[int]) -> int:
n = len(cost)
dp = [0]*(n)
dp[0] = cost[0]
dp[1] = cost[1]
for i in range(2,n):
dp[i] = min(dp[i-1], dp[i-2]) + cost[i]
return min(dp[n-1], dp[n-2])
时间复杂度:O(n)
空间复杂度:O(n)
这里同样可以只维持两个值就行了,从而使代码:
时间复杂度:O(n)
空间复杂度:O(1)
但为了直观,平时还是建议上述第一种写法。
力扣题目链接
1.一般解法,广度深度优先搜索,递归(超时):
可以转化为求二叉树叶子节点的个数
class Solution:
def uniquePaths(self, m: int, n: int) -> int:
def dfs(i, j, m, n):
# 越界
if i>m or j>n:
return 0
# 找到一种方法
if i==m and j==n:
return 1
# 递归,类似于走楼梯,而这里是往右或者往下走
return dfs(i+1, j, m, n) + dfs(i, j+1, m, n)
return dfs(1, 1, m, n)
2.动态规划:
确定dp数组(dp table)以及下标的含义
dp[i][j] :表示从(0, 0)出发,到(i, j) 有dp[i][j]条不同的路径。
确定递推公式
要求dp[i][j],得由前面的一个点推导而来,有两个,即上面的点dp[i - 1][j] 和 左边的点dp[i][j - 1]。
于是显而易见,公式为:dp[i][j] = dp[i - 1][j] + dp[i][j - 1],因为dp[i][j]只有这两个方向过来。
dp数组的初始化
如何初始化呢,首先dp[i][0]一定都是1,因为从(0, 0)的位置到(i, 0)的路径只有一条,那么dp[0][j]也同理。
所以初始化代码为:
for i in range(m):
dp[i][0] = 0
for i in range(n):
dp[0][j] = 0
确定遍历顺序
这里要看一下递归公式dp[i][j] = dp[i - 1][j] + dp[i][j - 1],dp[i][j]都是从其上方和左方推导而来,那么从左到右一层一层遍历就可以了。
这样就可以保证推导dp[i][j]的时候,dp[i - 1][j] 和 dp[i][j - 1]一定是有数值的。
class Solution:
def uniquePaths(self, m: int, n: int) -> int:
# 二维数组
dp = [[1]*n for _ in range(m)]
# 数组赋值
for i in range(1, m):
for j in range(1, n):
dp[i][j] = dp[i-1][j] + dp[i][j-1]
return dp[m-1][n-1]
时间复杂度:O(m × n)
空间复杂度:O(m × n)
同样的,在理解上述方法中表格的值的规律的情况下,可以只用维护一个一维数组,来减少空间复杂度,下一行等于上一行的与前一个点的和。
class Solution:
def uniquePaths(self, m: int, n: int) -> int:
dp = [1]*n
for i in range(1, m):
for j in range(1, n):
dp[j] += dp[j-1]
return dp[n-1]
时间复杂度:O(m × n)
空间复杂度:O(n)
一般这第二种同样也可能不便于理解,使用第一种方法即可。
力扣题目链接
动态规划:
动规五部曲:
所以需要加入条件判断:
# 当(i, j)没有障碍的时候,再推导dp[i][j]
if obstacleGrid[i][j] == 0:
dp[i][j] = dp[i - 1][j] + dp[i][j - 1]
# 或者
if obstacleGrid[i][j] == 1:
continue
dp[i][j] = dp[i - 1][j] + dp[i][j - 1];
dp = [[0]*n for _ in range(m)]
for i in range(m):
dp[0][i] = 1
for j in range(n):
dp[j][0] = 1
"""
虽然是初始化是:dp = [[1]*n for _ in range(m)],但这是简写,但实际上为上面的过程。
"""
但这道题,如上图,如果(i, 0) 这条边有了障碍之后,障碍之后(包括障碍)都是走不到的位置了,所以障碍之后的dp[i][0]应该还是初始值0。正确的初始化如下:
dp = [[0]*n for _ in range(m)]
for i in range(m):
if obstacleGrid[0][i] == 1:
break
dp[0][i] = 1
for j in range(n):
if obstacleGrid[j][0] == 1:
break
dp[j][0] = 1
注意代码里for循环的终止条件,一旦遇到obstacleGrid[i][0] == 1的情况就停止dp[i][0]的赋值1的操作,dp[0][j]同理。
for i in range(1, m):
for j in range(1, n):
if obstacleGrid[i][j] == 1:
continue
dp[i][j] = dp[i-1][j] + dp[i][j-1]
class Solution:
def uniquePathsWithObstacles(self, obstacleGrid: List[List[int]]) -> int:
length = len(obstacleGrid)
width = len(obstacleGrid[0])
# 如果障碍物在起点或终点,直接返回0
if obstacleGrid[0][0] == 1 or obstacleGrid[length-1][width-1]==1:
return 0
# 创建一个dp table
dp = [[0]*width for _ in range(length)]
# 两个循环,初始化 dp table
for i in range(length):
# 遇到障碍物,后续都为0
if obstacleGrid[i][0] == 1:
break
dp[i][0] = 1
for j in range(width):
# 遇到障碍物,后续都为0
if obstacleGrid[0][j] == 1:
break
dp[0][j] = 1
# 填表
for i in range(1, length):
for j in range(1, width):
if obstacleGrid[i][j] == 1:
continue
dp[i][j] = dp[i-1][j] + dp[i][j-1]
return dp[length-1][width-1]
时间复杂度:O(n × m),n、m 分别为obstacleGrid 长度和宽度
空间复杂度:O(n × m)
class Solution:
def uniquePathsWithObstacles(self, obstacleGrid: List[List[int]]) -> int:
length = len(obstacleGrid)
width = len(obstacleGrid[0])
# 如果障碍物在起点或终点,直接返回0
if obstacleGrid[0][0] == 1 or obstacleGrid[length-1][width-1]==1:
return 0
# 创建一个dp table
dp = [0]*width
# 一个循环,先从obstacleGrid第一行开始
for i in range(width):
# 遇到障碍物,后续都为0
if obstacleGrid[0][i] == 1:
break
dp[i] = 1
# 填表
# 从第二行开始
for i in range(1, length):
# 这里不能从1开始,因为可能位置0有障碍物,所以从第0列开始
for j in range(width):
# 有障碍物,走法数量直接为0
if obstacleGrid[i][j] == 1:
dp[j] = 0
# 这里不能直接else,否则会用下面公式累加而改变第一列的值
elif j!=0:
dp[j] += dp[j-1]
return dp[width-1]
时间复杂度:O(n × m),n、m 分别为obstacleGrid 长度和宽度
空间复杂度:O(m)
本题是 62.不同路径 的障碍版,整体思路大体一致。但就算是做过 62.不同路径,在做本题也会有感觉遇到障碍无从下手。其实只要考虑到,遇到障碍dp[i][j]保持0就可以了。也有一些小细节,例如:初始化的部分,很容易忽略了障碍之后应该都是0的情况。
64. 最小路径和
class Solution:
def minPathSum(self, grid: List[List[int]]) -> int:
row = len(grid)
col = len(grid[0])
dp = [[0]*col for _ in range(row)]
# 初始化边缘,这些都是固定值,无需判断的
dp[0][0]=grid[0][0]
for i in range(1, row):
dp[i][0]=grid[i][0]+dp[i-1][0]
for i in range(1, col):
dp[0][i]=grid[0][i]+dp[0][i-1]
for i in range(1, row):
for j in range(1, col):
# 取最小值就行
dp[i][j]=min(dp[i-1][j], dp[i][j-1])+grid[i][j]
return dp[-1][-1]
力扣题目链接
1 动态规划:
这道题求整数拆分后的最大乘积值的组合,使用动态规划来解,大的整数的拆分最大组合乘积等于其子整数的拆分最大组合乘积的乘积。
所以遍历顺序为:
for i in range(3, n+1):
for j in range(1, i//2+1):
dp[i] = max(dp[i], max(j*(i-j), j*dp[i-j]))
class Solution:
def integerBreak(self, n: int) -> int:
# 数值加上一,索引才会到n
dp = [0]*(n+1)
# 初始化,从2开始,因为k>=2
dp[2] = 1
# 拆分成k,k>=2,2前面已经初始化,从3开始
for i in range(3, n+1):
# 拆分成j 和 i-j, 例如:1*(3-1) = 2*(3-2)所以整除以一半,再加上一。
for j in range(1, i//2+1):
# 遍历j时,更新dp[i]的最大值
# 这里j*(i-j)是将i拆分成两部分,j*dp[i-j]是将i拆分成两部分以上
dp[i] = max(dp[i], max(j*(i-j), j*dp[i-j]))
# dp中最后一个索引为n,即将n拆分的最大乘积
return dp[-1]
时间复杂度:O(n^2)
空间复杂度:O(n)
2 数学证明:
本题也可以用贪心,每次拆成n个3,如果剩下是4,则保留4,然后相乘,但是这个结论需要数学证明其合理性,可以自行查阅研究。
class Solution:
def integerBreak(self, n: int) -> int:
if n==2:
return 1
if n==3:
return 2
if n==4:
return 4
result = 1
while n>4:
result *= 3
n -= 3
result *= n
return result
力扣题目链接
动态规划:
for i in range(1, n+1):
for j in range(1, i+1):
dp[i] += dp[j-1]*dp[i-j]
class Solution:
def numTrees(self, n: int) -> int:
# 从0到n
dp = [0]*(n+1)
dp[0] = 1
for i in range(1, n+1):
for j in range(1, i+1):
# 左子树数量*右子树数量
# 每种情况都要累加起来
dp[i] += dp[j-1]*dp[i-j]
return dp[-1]
时间复杂度: O ( n 2 ) O(n^2) O(n2)
空间复杂度: O ( n ) O(n) O(n)
背包问题,简单来说就是将单次、无限重复或者有限重复数量的物品,通过满足一定的题目需求,而放入有限大小的背包中。
而根据物品的三种不同规则,我们将分为三种背包问题:01背包,完全背包和多重背包。
它们的总结如下图:
但我们可以看到还有其他类型的背包在其中,但对于面试的话,其实掌握01背包,和完全背包,就够用了,最多可以再来一个多重背包。
至于其他的背包问题,面试几乎不会问,都是竞赛级别的了,leetcode上连多重背包的题目都没有,所以题库也告诉我们,01背包和完全背包就够用了。但如果你热爱学习,喜欢钻研,则一本背包问题的经典书籍送给你 背包问题九讲 2.0。
理解背包问题,01背包非常基础和重要,因为其他问题基本由此为基础衍生而来。我们先从纯背包问题开始,去理解和学习,再去延伸和转化到leetcode题目。
一次数量的物品,根据要求放入背包中。
暴力解法:
每一件物品其实只有两个状态:取或者不取
,所以可以使用回溯法搜索出所有的情况,那么时间复杂度就是 o ( 2 n ) o(2^n) o(2n),这里的n表示物品数量。
所以暴力的解法是指数级别的时间复杂度。进而才需要动态规划的解法来进行优化!
问题描述: 有 n n n件物品和一个最多能背重量为 w w w的背包。第 i i i件物品的重量是 w e i g h t [ i ] weight[i] weight[i],得到的价值是 v a l u e [ i ] value[i] value[i] 。每件物品只能用一次,求解将哪些物品装入背包里物品价值总和最大。
简单例子:
有 w = 4 w=4 w=4重量的背包
有 n = 3 n=3 n=3件物品,表示如下:
重量 weight[i] | 价值 value[i] | |
---|---|---|
物品0 | 1 | 15 |
物品1 | 3 | 20 |
物品2 | 4 | 30 |
问背包能背的物品的最大价值是多少?
(1). 二维DP数组
动态规划五部曲
确定dp数组以及下标的含义
对于背包问题,有一种写法, 是使用二维数组,即 d p [ i ] [ j ] dp[i][j] dp[i][j] 表示从下标为 [ 0 − i ] [0-i] [0−i]的物品里任意取,放进容量为j的背包,价值总和最大是多少。
确定递归公式
d p [ i ] [ j ] dp[i][j] dp[i][j]的含义为,从下标为 0 − i 0-i 0−i的物品里任取,放进容量为j的背包,价值总和最大是多少。
从两个方向可以推出 d p [ i ] [ j ] dp[i][j] dp[i][j]:
数组的初始化
首先从 d p [ i ] [ j ] dp[i][j] dp[i][j]的定义出发,如果背包容量 j j j为 0 0 0的话,即 d p [ i ] [ 0 ] dp[i][0] dp[i][0],无论是选取哪些物品,背包价值总和一定为 0 0 0。如图:
再看其他情况,状态转移方程 d p [ i ] [ j ] = m a x ( d p [ i − 1 ] [ j ] , d p [ i − 1 ] [ j − w e i g h t [ i ] ] + v a l u e [ i ] ) dp[i][j] = max(dp[i-1][j], dp[i-1][j-weight[i]]+value[i]) dp[i][j]=max(dp[i−1][j],dp[i−1][j−weight[i]]+value[i]),因为 i i i是由 i − 1 i-1 i−1推导而来,所以 i = 0 i=0 i=0一定要初始化。
即初始化 d p [ 0 ] [ j ] dp[0][j] dp[0][j],存放编号为0的物品时,各个容量的背包所能存放的最大值。
于是:
for j in range(bagweight):
if j>=weight[0]:
dp[0][j]=value[0]
else:
dp[0][j]=0
至于其他值,都会被递推公式推出结果,直接初始化微为即可,初始化完成后为:
4. 确定遍历顺序
有两个遍历的维度:物品与背包重量。
先遍历 物品还是先遍历背包重量呢?两者皆可,但是先遍历物品更好理解,比较贴合上面的图。
# 物品的编号
for i in range(1, len(weight)):
# 包的大小
for j in range(1, bagweight):
# 大于包的大小,不放,与上一个物品价值相等
if j<weight[i]:
dp[i][j] = dp[i-1][j]
# 不放与放取较大的值
else:
dp[i][j] = max(dp[i-1][j], dp[i-1][j-weight[i]]+value[i])
整体代码(二维):
def bagpack_problem1(bag_size, weight, value) -> int:
# rows 物品 cols 背包
rows, cols = len(weight), bag_size + 1
# 这里先遍历物品,再遍历背包。
# 若先遍历背包,再遍历物品,下面dp表的创建行列要互换,初始化互换,下面的两个循环也要互换
dp = [[0 for _ in range(cols)] for _ in range(rows)]
# 初始化dp数组
# 第0列,0-rows行,因为背包大小为0,则直接为0
for i in range(rows):
dp[i][0] = 0
first_item_weight, first_item_value = weight[0], value[0]
# 第0行,1-rows列(因为0在前面已经赋值为0了,所以从1开始),小于等于背包大小的物品赋予物品大小的值,其余为0
for j in range(1, cols):
if first_item_weight <= j:
dp[0][j] = first_item_value
# 更新dp数组: 先遍历物品, 再遍历背包.
for i in range(1, len(weight)):
cur_weight, cur_val = weight[i], value[i]
# 注意这里能写成for j in range(nums[i],target+1)这样吗?
# 不行,因为前面的值需要if cur_weight > j 去将0填空,便于下一行的值的依赖
# 所以这里不同于二维数组,不需要依赖,而直接到nums[i]为止
for j in range(1, cols):
if cur_weight > j: # 说明背包装不下当前物品.
dp[i][j] = dp[i - 1][j] # 所以不装当前物品.
else:
# 定义dp数组: dp[i][j] 前i个物品里,放进容量为j的背包,价值总和最大是多少。
dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - cur_weight]+ cur_val)
print(dp)
if __name__ == "__main__":
bag_size = 4
weight = [1, 3, 4]
value = [15, 20, 30]
bagpack_problem1(bag_size, weight, value)
(2). 一维DP数组(滚动数组)
背包问题的状态其实都是可以压缩的。
在使用二维数组的时候,递推公式: d p [ i ] [ j ] = m a x ( d p [ i − 1 ] [ j ] , d p [ i − 1 ] [ j − w e i g h t [ i ] ] + v a l u e [ i ] ) dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - weight[i]] + value[i]) dp[i][j]=max(dp[i−1][j],dp[i−1][j−weight[i]]+value[i])
如果把 d p [ i − 1 ] dp[i - 1] dp[i−1]那一层拷贝到 d p [ i ] dp[i] dp[i]上,表达式完全可以是: d p [ i ] [ j ] = m a x ( d p [ i ] [ j ] , d p [ i ] [ j − w e i g h t [ i ] ] + v a l u e [ i ] ) dp[i][j] = max(dp[i][j], dp[i][j - weight[i]] + value[i]) dp[i][j]=max(dp[i][j],dp[i][j−weight[i]]+value[i])
与其把 d p [ i − 1 ] dp[i - 1] dp[i−1]这一层拷贝到 d p [ i ] dp[i] dp[i]上,不如只用一个一维数组了,只用 d p [ j ] dp[j] dp[j](一维数组,也可以理解是一个滚动数组)。
这就是滚动数组的由来,需要满足的条件是上一层可以重复利用,直接拷贝到当前层。
动态规划五部曲
dp[j] = max(dp[j], dp[j - weight[i]] + value[i]);
for i in range(len(weight)):
for j in reversed(range(weight[i], bagweight+1)):
dp[j] = max(dp[j], dp[j-weight[i]]+value[i])
这里大家发现和二维dp的写法中,遍历背包的顺序是不一样的!
二维dp遍历的时候,背包容量是从小到大,而一维dp遍历的时候,背包是从大到小。
为什么呢?
倒序遍历是为了保证物品i只被放入一次!。但如果一旦正序遍历了,那么物品0就会被重复加入多次!
举一个例子:物品0的重量weight[0] = 1,价值value[0] = 15
如果正序遍历
dp[1] = dp[1 - weight[0]] + value[0] = 15
dp[2] = dp[2 - weight[0]] + value[0] = 30
此时dp[2]就已经是30了,意味着物品0,被放入了两次,所以不能正序遍历。
为什么倒序遍历,就可以保证物品只放入一次呢?
倒序就是先算dp[2]
dp[2] = dp[2 - weight[0]] + value[0] = 15 (dp数组已经都初始化为0)
dp[1] = dp[1 - weight[0]] + value[0] = 15
所以从后往前循环,每次取得状态不会和之前取得状态重合,这样每种物品就只取一次了。
那么问题又来了,为什么二维dp数组历的时候不用倒序呢?
因为对于二维dp,dp[i][j]都是通过上一层即dp[i - 1][j]计算而来,本层的dp[i][j]并不会被覆盖!
再来看看两个嵌套for循环的顺序,代码中是先遍历物品嵌套遍历背包容量,那可不可以先遍历背包容量嵌套遍历物品呢?不可以!
因为一维dp的写法,背包容量一定是要倒序遍历(原因上面已经讲了),如果遍历背包容量放在上一层,那么每个dp[j]就只会放入一个物品,即:背包里只放入了一个物品。
倒序遍历的原因是,本质上还是一个对二维数组的遍历,并且右下角的值依赖上一层左上角的值,因此需要保证左边的值仍然是上一层的,从右向左覆盖。
(这里如果读不懂,就再回想一下dp[j]的定义,或者就把两个for循环顺序颠倒一下试试!)
所以一维dp数组的背包在遍历顺序上和二维其实是有很大差异的!,这一点大家一定要注意。
整体代码(一维):
def bag_problem():
weight = [1, 3, 4]
value = [15, 20, 30]
bag_weight = 4
# 初始化: 全为0
dp = [0] * (bag_weight + 1)
# 先遍历物品, 再遍历背包容量
for i in range(len(weight)):
for j in range(bag_weight, weight[i] - 1, -1):
# 递归公式
dp[j] = max(dp[j], dp[j - weight[i]] + value[i])
print(dp)
bag_problem()
总结:
动态规划问题枚举的顺序只有一个原则:如果b状态的计算依赖于a状态,那么a状态必须在b状态之前被枚举和计算。
二维数组时,物品和背包的顺序,从小到大遍历,二者可以内外互换,物品顺序可颠倒,但初始化操作也要相应颠倒;而背包只可逆序,因为背包的结果与前面一次较小的背包结果有关,而前面的结果是上一层的拷贝,而不会受到这一层更新的干扰而重复累加,避免重复计算,重复计算会变成完全背包。
对于二维dp,dp[i][j]都是通过上一层即dp[i - 1][j]计算而来,所以二维无需逆序遍历,当然背包也可以逆序,不影响结果。
01背包最大重量/价值 | 二维 | 一维 |
---|---|---|
初始化 | 第0行0列初始化 | 无需初始化 |
内外循序 | 物品+背包 | 背包+物品 | 物品+背包 |
物品顺序 | 正序|逆序 | 正序|逆序 |
背包顺序 | 正序|逆序 | 逆序 |
递推公式 | dp[i][j]=max(dp[i-1]j[j], dp[i-1][j-weight[i]]+value[i]) | dp[j]=max(dp[j], dp[j-weight[i]]+value[i]) |
问题描述: 有 n n n件物品和一个最多能背重量为 w w w的背包。第i件物品的重量是 w e i g h t [ i ] weight[i] weight[i],每件物品只能用一次,求解将哪些物品装入背包里物品重量总和最大。
分析:
这个问题没有给每个物品的价值,也没有问最多能获得多少价值,而是问最多能装多满。
实际上在这里,如果我们把每个物品的大小当作每个物品的价值,就可以完美解决这个问题。
简单例子:
有 w = 4 w=4 w=4重量的背包
有 n = 3 n=3 n=3件物品,表示如下:
这里增设一列不存在的价值,值与重量相等,模拟等于上一题求最大价值。
重量 weight[i] | 价值 value[i] | |
---|---|---|
物品0 | 1 | 1 |
物品1 | 3 | 3 |
物品2 | 4 | 4 |
问背包能背的物品的最大价值(或者重量)是多少?
代码同上,只不过values与weights相等,那么只需把values改成weight即可。
整体代码(一维,滚动数组):
def bag_problem():
weight = [1, 3, 4]
value = [15, 20, 30]
bag_weight = 4
# 初始化: 全为0
dp = [0] * (bag_weight + 1)
# 先遍历物品, 再遍历背包容量
for i in range(len(weight)):
for j in range(bag_weight, weight[i] - 1, -1):
# 递归公式,这里注意不是+value[i]而是weight[i]
dp[j] = max(dp[j], dp[j - weight[i]] + weight[i])
print(dp)
bag_problem()
416. 分割等和子集
解题思路: 等分,先判断总和能否被2整除,能否进行分隔,不能直接False,能的话继续判断里面的数,分隔后能否分成相等的两部分,等分必然是总和值/2的值的大小的背包装满,能装满则说明能等分,不能装满则False。
二维数组:
class Solution:
def canPartition(self, nums: List[int]) -> bool:
# target既是背包大小也是最大价值,问是否能装满target大小的背包
target = sum(nums)
# 如果是奇数则不能均分,直接False,如果偶数则再后续加以判断
if target%2==1:
return False
else:
target //=2
dp = [[0]*(target+1) for _ in range(len(nums))]
# 遍历物品
for i in range(len(nums)):
# 遍历背包大小
for j in range(target+1):
# 背包小鱼物品大小
if j<nums[i]:
dp[i][j] = dp[i-1][j]
# 能放但不放,或者放进去那个大取哪个
else:
dp[i][j] = max(dp[i-1][j], dp[i-1][j-nums[i]]+nums[i])
return target == dp[-1][-1]
一维数组(滚动数组):
class Solution:
def canPartition(self, nums: List[int]) -> bool:
# target既是背包大小也是最大价值,问是否能装满target大小的背包
target = sum(nums)
# 如果是奇数则不能均分,直接False,如果偶数则再后续加以判断
if target%2==1:
return False
else:
target //=2
dp = [0]*(target+1)
# 遍历物品
for i in range(len(nums)):
# 遍历背包大小,逆序防止累加
# for j in reversed(range(nums[i], target+1)):
for j in range(target, nums[i]-1, -1):
dp[j] = max(dp[j], dp[j-nums[i]]+nums[i])
return target == dp[-1]
1049. 最后一块石头的重量 II
该题与上一题很类似,隐藏的背包问题。最终的思路是将这个石头集合分成两部分,他们越相近,即离和中值越近,相差就越小。如果无法等分,那么因为这个中值为向下取整,所以离偏小的部分近一些,离偏大的部分远一些。那么以中值为背包,装满得到较小值部分的值,求石头的总体和,减去较小值部分,就为最大值部分,再减去一次较小值部分,得出最小的相差。
为什么不求较大值呢,因为较小值可以用中值找到最大背包,而较大值的最大背包找不到。
一维滚动数组:
class Solution:
def lastStoneWeightII(self, stones: List[int]) -> int:
su_m = sum(stones)
target = su_m//2
dp = [0]*(target+1)
for i in range(len(stones)):
for j in range(target, stones[i]-1,-1):
dp[j] = max(dp[j], dp[j-stones[i]]+stones[i])
return su_m-dp[-1]-dp[-1]
474. 一和零
二维背包问题,分别装0和1的数量的背包,两个数量用一个二维来存储。
class Solution:
def findMaxForm(self, strs: List[str], m: int, n: int) -> int:
dp = [[0]*(n+1) for _ in range(m+1)]
# 遍历物品
for s in strs:
ones = s.count('1')
zeros = s.count('0')
# 遍历二维背包
for i in range(m, zeros-1, -1):
for j in range(n, ones-1, -1):
dp[i][j] = max(dp[i][j], dp[i-zeros][j-ones]+1)
return dp[-1][-1]
问题描述: 有 n n n件物品和一个最多能背重量为 w w w的背包。第i件物品的重量是 w e i g h t [ i ] weight[i] weight[i],每件物品只能用一次,求解装满背包有哪些不同的方法。
分析:
这次和之前遇到的背包问题不一样了,之前都是求容量为j的背包,最多能装多少。
本题则是装满有几种方法。其实这就是一个组合问题了。求组合类问题的公式,都是类似这种:
1. 确定dp[j]的含义:
dp[j] 表示:填满j(包括j)这么大容积的包,有dp[j]种方法
2. 递推公式:
只要获得nums[i],凑成dp[j]就有dp[j - nums[i]] 种方法。
例如:dp[j],j 为5,
如果 | 那么 |
---|---|
已经有一个1(nums[i]) 的话 | 有 dp[4]种方法 凑成 容量为5的背包 |
已经有一个2(nums[i]) 的话 | 有 dp[3]种方法 凑成 容量为5的背包 |
已经有一个3(nums[i]) 的话 | 有 dp[2]中方法 凑成 容量为5的背包 |
已经有一个4(nums[i]) 的话 | 有 dp[1]中方法 凑成 容量为5的背包 |
已经有一个5 (nums[i])的话 | 有 dp[0]中方法 凑成 容量为5的背包 |
那么凑整dp[5]有多少方法呢,也就是把 所有的 dp[j - nums[i]] 累加起来。
dp[j] += dp[j - nums[i]]
这个公式在后面在讲解背包解决排列组合问题的时候还会用到。
3. 初始化:
dp数组如何初始化?
从递推公式可以看出,在初始化的时候dp[0] 一定要初始化为1,因为dp[0]是在公式中一切递推结果的起源,如果dp[0]是0的话,递推结果将都是0。
4. 遍历顺序:
01背包问题一维dp的遍历,nums放在外循环,target在内循环,且内循环倒序。
简单例子:
有 w = 4 w=4 w=4重量的背包
有 n = 3 n=3 n=3件物品,表示如下:
重量 weight[i] | |
---|---|
物品0 | 2 |
物品1 | 2 |
物品2 | 4 |
问有多少种装满背包的方法?
结果:[2, 2], [4]
494. 目标和
将该题转化为背包装满的装法次数。
使用推导公式:
dp[j] += dp[j-nums[i]]
背包大小根据公式推导而来
left+right = sum
left-right = target
left = (target+sum)/2
初始化时,需要赋值第一个值为1,其余为0不变
dp[0]=1
举例推导:
输入:nums: [1, 1, 1, 1, 1], S: 3
bagSize = (S + sum) / 2 = (3 + 5) / 2 = 4
代码整体如下:
class Solution:
def findTargetSumWays(self, nums: List[int], target: int) -> int:
# left+right = sum
# left-right = target
# left = (target+sum)/2
s_um = sum(nums)
# 目标值比所有值求和大,或者目标值不可分,那么方法为0
if abs(target)>s_um or (target+s_um)%2:
return 0
# 背包大小,根据上面公式推导
bagsize = (target+s_um)//2
# 创建dp表
dp = [0]*(bagsize+1)
# 初始化第一个元素为1
dp[0]=1
# 同前面的二重循环
for i in range(len(nums)):
for j in range(bagsize, nums[i]-1, -1):
# 递推公式,用来求背包装满的次数
dp[j] += dp[j-nums[i]]
return dp[-1]
注意:
有道题 39. 组合总和 大部分是使用回溯算法来做的,为什么不选用动态规划呢?因为题目所求的结果不同:
求个数 | 列出组合 |
---|---|
动态规划 | 回溯爆搜 |
总结:
01背包最大重量/价值 | 01背包最大重量/价值 | 01背包可能性总数 | |
---|---|---|---|
维度 | 二维 | 一维 | 一维 |
初始化 | 第0行和第0列分别初始化 | 无需初始化 | 初始化dp[0]=1 |
内外循序 | 物品+背包 | 背包+物品 | 物品+背包 | 物品+背包 |
物品顺序 | 正序|逆序 | 正序|逆序 | 正序|逆序 |
背包顺序 | 正序|逆序 | 逆序 | 逆序 |
递推公式 | dp[i][j]=max(dp[i-1]j[j], dp[i-1][j-weight[i]]+value[i]) | dp[j]=max(dp[j], dp[j-weight[i]]+value[i]) | dp[j]+=dp[j-nums[i]] |
完全背包和01背包问题唯一不同的地方就是,每种物品有无限件。
问题描述: 有N件物品和一个最多能背重量为W的背包。第i件物品的重量是weight[i],得到的价值是value[i] 。每件物品都有无限个(也就是可以放入背包多次),求解将哪些物品装入背包里物品价值总和最大。
简单例子:
有 w = 4 w=4 w=4重量的背包
有 n = ∞ n=\infty n=∞件物品,表示如下:
重量 weight[i] | 价值 value[i] | |
---|---|---|
物品0 | 1 | 15 |
物品1 | 3 | 20 |
物品2 | 4 | 30 |
问背包能背的物品的最大价值是多少?
对比下:
01背包的核心代码如下:
# 先遍历物品,再遍历背包
# 遍历物品
for i in range(len(weight)):
# 遍历背包容量
for j in range(bagWeight, weight[i]-1, -1):
dp[j] = max(dp[j], dp[j - weight[i]]+value[i])
我们知道01背包内嵌的循环是从大到小遍历,为了保证每个物品仅被添加一次。
而完全背包的物品是可以添加多次的,所以要从小到大去遍历,即:
# 先遍历物品,再遍历背包
# 遍历物品
for i in range(len(weight)):
# 遍历背包容量
for j in range(weight[i], bagWeight+1):
dp[j] = max(dp[j], dp[j-weight[i]]+value[i])
01背包中二维dp数组的两个for遍历的先后循序是可以颠倒了,一维dp数组的两个for循环先后循序一定是先遍历物品,再遍历背包容量。
在完全背包中,对于一维dp数组来说,其实两个for循环嵌套顺序是无所谓的
。
因为dp[j] 是根据 下标j之前所对应的dp[j]计算出来的。 只要保证下标j之前的dp[j]都是经过计算的就可以了。
遍历物品在外层循环,遍历背包容量在内层循环,状态如图:
遍历背包容量在外层循环,遍历物品在内层循环,状态如图:
看了这两个图,大家就会理解,完全背包中,两个for循环的先后循序,都不影响计算dp[j]所需要的值(这个值就是下标j之前所对应的dp[j])。
先遍历背包再遍历物品,代码如下:
# 先遍历背包,再遍历物品
# 遍历背包
for j in range(bagWeight+1):
# 遍历物品
for i in range(len(weight)):
if j-weight[i]>=0:
dp[j] = max(dp[j-weight[i]]+value[i])
整体代码:
# 先遍历物品,再遍历背包
def bag_pack1():
weight = [1, 3, 4]
value = [15, 20, 30]
bag_weight = 4
dp = [0]*(bag_weight + 1)
for i in range(len(weight)):
for j in range(weight[i], bag_weight + 1):
dp[j] = max(dp[j], dp[j - weight[i]] + value[i])
print(dp[bag_weight])
# 先遍历背包,再遍历物品
def bag_pack2():
weight = [1, 3, 4]
value = [15, 20, 30]
bag_weight = 4
dp = [0]*(bag_weight + 1)
for j in range(bag_weight + 1):
for i in range(len(weight)):
if j >= weight[i]:
dp[j] = max(dp[j], dp[j - weight[i]] + value[i])
print(dp[bag_weight])
if __name__ == '__main__':
bag_pack1()
bag_pack2()
总结:
因为是完全背包,物品可叠加,那么一维数组顺序遍历即可:
01背包最大重量/价值 | 01背包最大重量/价值 | 01背包可能性总数 | 完全背包最大重量/价值 | |
---|---|---|---|---|
维度 | 二维 | 一维 | 一维 | 一维 |
初始化 | 第0行和第0列分别初始化 | 无需初始化 | 初始化dp[0]=1 | 无需初始化 |
内外循序 | 物品+背包 | 背包+物品 | 物品+背包 | 物品+背包 | 物品+背包|背包+物品 |
物品顺序 | 正序|逆序 | 正序|逆序 | 正序|逆序 | 正序|逆序 |
背包顺序 | 正序|逆序 | 逆序(避免重复) | 逆序 | 正序(为了重复) |
递推公式 | dp[i][j]=max(dp[i-1]j[j], dp[i-1][j-weight[i]]+value[i]) | dp[j]=max(dp[j], dp[j-weight[i]]+value[i]) | dp[j]+=dp[j-nums[i]] | dp[i][j]=max(dp[i-1]j[j], dp[i-1][j-weight[i]]+value[i]) |
518. 零钱兑换 II
递推公式:
求背包装满的次数,根据公式:dp[j] = dp[j-nums[i]]
遍历顺序:
因为纯完全背包求得装满背包的最大价值是多少,和凑成总和的元素有没有顺序没关系,即:有顺序也行,没有顺序也行。所以纯完全背包是能凑成总和就行,不用管怎么凑的。
而本题要求凑成总和的组合数,元素之间明确要求没有顺序,那么每个方案个数是为组合数。
那么本题,两个for循环的先后顺序可就有说法了。
结果是求组合数,那么遍历顺序为:
先遍历物品,后遍历背包:
# 遍历物品
for i in range(len(coins)):
# 遍历背包容量
for j in range(coins[i], amount+1):
dp[j] += dp[j-coins[i]]
如果是求排列数,那么遍历顺序为:
# 遍历背包容量
for j in range(amount+1):
# 遍历物品
for i in range(len(coins)):
if j-coins[i]>=0:
dp[j]+=dp[j-coins[i]]
整体代码:
class Solution:
def change(self, amount: int, coins: List[int]) -> int:
dp = [0]*(amount+1)
dp[0]=1
for i in range(len(coins)):
for j in range(coins[i], amount+1):
dp[j]+=dp[j-coins[i]]
return dp[-1]
总结:
通过该例子,我们知道,在完全背包问题的装满可能性总数中,物品和背包的顺序在排列和组合问题中是有影响的:
排列 | 组合 |
---|---|
背包+物品 | 物品+背包 |
综上:
01背包最大重量/价值 | 01背包最大重量/价值 | 01背包可能性总数 | 完全背包最大重量/价值 | 完全背包可能性总数 | |
---|---|---|---|---|---|
维度 | 二维 | 一维 | 一维 | 一维 | 一维 |
初始化 | 第0行和第0列分别初始化 | 无需初始化 | 初始化dp[0]=1 | 无需初始化 | 初始化dp[0]=1 |
内外循序 | 物品+背包 | 背包+物品 | 物品+背包 | 物品+背包 | 物品+背包|背包+物品 | 排列: 背包+物品|组合: 物品+背包 |
物品顺序 | 正序|逆序 | 正序|逆序 | 正序|逆序 | 正序|逆序 | 正序|逆序 |
背包顺序 | 正序|逆序 | 逆序(避免重复) | 逆序 | 正序(为了重复) | 正序 |
递推公式 | dp[i][j]=max(dp[i-1]j[j], dp[i-1][j-weight[i]]+value[i]) | dp[j]=max(dp[j], dp[j-weight[i]]+value[i]) | dp[j]+=dp[j-nums[i]] | dp[i][j]=max(dp[i-1]j[j], dp[i-1][j-weight[i]]+value[i]) | dp[j]+=dp[j-nums[i]] |
377. 组合总和 Ⅳ
完全背包,两个循环顺序执行
求次数,dp[j] = dp[j-nums[i]],并且dp[0]=1
求排列,先背包后物品。
根据上一题的规律,可以很轻易的做出此题:
class Solution:
def combinationSum4(self, nums: List[int], target: int) -> int:
dp = [0]*(target+1)
dp[0]=1
for j in range(target+1):
for i in nums:
if j-i>=0:
dp[j]+=dp[j-i]
return dp[-1]
70. 爬楼梯
同上题的思路,这是背包里求排列问题,即:1、2 步 和 2、1 步都是上三个台阶,但是这两种方法不一样。所以需将target放在外循环,将nums放在内循环。每一步可以走多次,这是完全背包,内循环需要从前向后遍历。
class Solution:
def climbStairs(self, n: int) -> int:
dp = [0]*(n+1)
dp[0]=1
# 遍历背包数,楼梯总长度
for j in range(n+1):
# 遍历物品,台阶数,1或2
for i in [1,2]:
if j-i>=0:
dp[j]+=dp[j-i]
return dp[-1]
139. 单词拆分
1.dp数组含义
dp[i] : 字符串长度为i的话,dp[i]为true,表示可以拆分为一个或多个在字典中出现的单词。
2.递推公式
如果确定dp[j] 是true,且 [j, i] 这个区间的子串出现在字典里,那么dp[i]一定是true。(j < i )。
所以递推公式是 if([j, i] 这个区间的子串出现在字典里 && dp[j]是true) 那么 dp[i] = true。
3.dp数组的初始化
从递推公式中可以看出,dp[i] 的状态依靠 dp[j]是否为true,那么dp[0]就是递推的根基,dp[0]一定要为true,否则递推下去后面都都是false了。
下标非0的dp[i]初始化为false,只要没有被覆盖说明都是不可拆分为一个或多个在字典中出现的单词。
4.遍历顺序
题目中说是拆分为一个或多个在字典中出现的单词,所以这是完全背包。
“apple”, “pen” 是物品,那么我们要求 物品的组合一定是 “apple” + “pen” + “apple” 才能组成 “applepenapple”。
“apple” + “apple” + “pen” 或者 “pen” + “apple” + “apple” 是不可以的,那么我们就是强调物品之间顺序。
所以说,本题一定是 先遍历 背包,再遍历物品。所以是排列问题。
5.举例推导dp[i]
以输入: s = “leetcode”, wordDict = [“leet”, “code”]为例,dp状态如图:
所以整体代码如下:
class Solution:
def wordBreak(self, s: str, wordDict: List[str]) -> bool:
dp = [False]*(len(s)+1)
dp[0]=True
# 排列问题,先背包
for j in range(1, len(s)+1):
# 后物品,即单词
for i in wordDict:
if j>=len(i):
# or前的dp[j]为True时防止被后面为False的覆盖
# or后的部分判断是否有True,即匹配的单词
dp[j] = dp[j] or dp[j-len(i)] and s[j-len(i):j]==i
return dp[-1]
322. 零钱兑换
1.dp数组的含义:
dp[j]:凑足总额为j所需钱币的最少个数为dp[j]
2.递推公式:
凑足总额为j - coins[i]的最少个数为dp[j - coins[i]],那么只需要加上一个钱币coins[i]即dp[j - coins[i]] + 1就是dp[j](考虑coins[i])所以dp[j] 要取所有 dp[j - coins[i]] + 1 中最小的。
求最小次数:dp[j] = min(dp[j], dp[j - coin[i]] + 1)
3.数组初始化
dp数组如何初始化
首先凑足总金额为0所需钱币的个数一定是0,那么dp[0] = 0;
其他下标对应的数值呢?考虑到递推公式的特性,如果默认初始化为0,那么每次求min的时候,都会取默认值0而不是真实最小值,那么默认0就不可取。所以dp[j]必须初始化为一个最大的数,否则就会在min(dp[j - coins[i]] + 1, dp[j])比较的过程中被初始值覆盖。所以下标非0的元素都是应该是最大值。
代码如下:
dp = [float('inf')]*(amount+1)
dp[0] = 1
4.遍历顺序
求所需钱币最小个数,所以并不讲究物品和背包的先后顺序,因为既不是排列也不是组合。
5.举例推导dp数组
以输入:coins = [1, 2, 5], amount = 5为例:
整理代码如下:
class Solution:
def coinChange(self, coins: List[int], amount: int) -> int:
'''版本一'''
# 初始化
dp = [float("inf")]*(amount + 1)
dp[0] = 0
# 遍历物品
for coin in coins:
# 遍历背包
for j in range(coin, amount + 1):
dp[j] = min(dp[j], dp[j - coin] + 1)
return dp[amount] if dp[amount] != float("inf") else -1
def coinChange1(self, coins: List[int], amount: int) -> int:
'''版本二'''
# 初始化
dp = [float("inf")]*(amount + 1)
dp[0] = 0
# 遍历物品
for j in range(1, amount + 1):
# 遍历背包
for coin in coins:
if j >= coin:
dp[j] = min(dp[j], dp[j - coin] + 1)
return dp[amount] if dp[amount] != float("inf") else -1
总结:
01背包最大重量/价值 | 01背包最大重量/价值 | 01背包可能性总数 | 完全背包最大重量/价值 | 完全背包可能性总数 | 完全背包最少数量 | |
---|---|---|---|---|---|---|
维度 | 二维 | 一维 | 一维 | 一维 | 一维 | 一维 |
初始化 | 第0行和第0列分别初始化 | 无需初始化 | 初始化dp[0]=1 | 无需初始化 | 初始化dp[0]=1 | dp = [float(“inf”)]*(amount + 1) dp[0] = 0 |
内外循序 | 物品+背包 | 背包+物品 | 物品+背包 | 物品+背包 | 物品+背包|背包+物品 | 排列: 背包+物品|组合: 物品+背包 | 背包+物品|物品+背包 |
物品顺序 | 正序|逆序 | 正序|逆序 | 正序|逆序 | 正序|逆序 | 正序|逆序 | 正序 |
背包顺序 | 正序|逆序 | 逆序(避免重复) | 逆序 | 正序(为了重复) | 正序 | 正序 |
递推公式 | dp[i][j]=max(dp[i-1]j[j], dp[i-1][j-weight[i]]+value[i]) | dp[j]=max(dp[j], dp[j-weight[i]]+value[i]) | dp[j]+=dp[j-nums[i]] | dp[i][j]=max(dp[i-1]j[j], dp[i-1][j-weight[i]]+value[i]) | dp[j]+=dp[j-nums[i]] | dp[j] = min(dp[j], dp[j - coin[i]] + 1) |
279. 完全平方数
这题同上一题零钱兑换,思路类似,不过得自创平方数集合作为物品。
class Solution:
def numSquares(self, n: int) -> int:
nums = [i*i for i in range(1, n+1) if i*i<=n]
dp = [float('inf')]*(n+1)
dp[0]=0
# 先遍历物品,再遍历背包
for i in nums:
for j in range(i, n+1):
dp[j] = min(dp[j], dp[j-i]+1)
return dp[-1]
也可以先遍历背包,再遍历物品
# 遍历背包
for j in range(1, n + 1):
# 遍历物品
for num in nums:
if j >= num:
dp[j] = min(dp[j], dp[j - num] + 1)
这个问题与 0-1 背包的区别在于,0-1 背包中每种物品有且只有一个,而多重背包中一种物品有nums[i] 个。
对于多重背包,我在力扣上还没发现对应的题目,所以这里就做一下简单介绍,大家大概了解一下。
有N种物品和一个容量为V 的背包。第i种物品最多有Mi件可用,每件耗费的空间是Ci ,价值是Wi 。求解将哪些物品装入背包可使这些物品的耗费的空间 总和不超过背包容量,且价值总和最大。
多重背包和01背包是非常像的, 为什么和01背包像呢?
每件物品最多有Mi件可用,把Mi件摊开,其实就是一个01背包问题了。
例如:
背包最大重量为10。
物品为:
重量 weight[i] | 价值 value[i] | 数量 | |
---|---|---|---|
物品0 | 1 | 15 | 2 |
物品1 | 3 | 20 | 3 |
物品2 | 4 | 30 | 2 |
问背包能背的物品最大价值是多少?
和如下情况有区别么?
重量 weight[i] | 价值 value[i] | 数量 | |
---|---|---|---|
物品0 | 1 | 15 | 1 |
物品0 | 1 | 15 | 1 |
物品1 | 3 | 20 | 1 |
物品1 | 3 | 20 | 1 |
物品1 | 3 | 20 | 1 |
物品2 | 4 | 30 | 1 |
物品2 | 4 | 30 | 1 |
毫无区别,这就转成了一个01背包问题了,且每个物品只用一次。 |
这种方式来实现多重背包的代码如下:
def test_multi_pack1():
'''版本一:改变物品数量为01背包格式'''
weight = [1, 3, 4]
value = [15, 20, 30]
nums = [2, 3, 2]
bag_weight = 10
for i in range(len(nums)):
# 将物品展开数量为1
while nums[i] > 1:
weight.append(weight[i])
value.append(value[i])
nums[i] -= 1
dp = [0]*(bag_weight + 1)
# 遍历物品
for i in range(len(weight)):
# 遍历背包
for j in range(bag_weight, weight[i] - 1, -1):
dp[j] = max(dp[j], dp[j - weight[i]] + value[i])
print(" ".join(map(str, dp))
if __name__ == '__main__':
test_multi_pack1()
时间复杂度:O(m × n × k),m:物品种类个数,n背包容量,k单类物品数量
也有另一种实现方式,就是把每种商品遍历的个数放在01背包里面在遍历一遍。
代码如下:(详看注释)
def test_multi_pack2():
'''版本:改变遍历个数'''
weight = [1, 3, 4]
value = [15, 20, 30]
nums = [2, 3, 2]
bag_weight = 10
dp = [0]*(bag_weight + 1)
for i in range(len(weight)):
for j in range(bag_weight, weight[i] - 1, -1):
# 以上是01背包,加上遍历个数
for k in range(1, nums[i] + 1):
if j - k*weight[i] >= 0:
dp[j] = max(dp[j], dp[j - k*weight[i]] + k*value[i])
print(" ".join(map(str, dp)))
if __name__ == '__main__':
test_multi_pack2()
时间复杂度:O(m × n × k),m:物品种类个数,n背包容量,k单类物品数量
从代码里可以看出是01背包里面在加一个for循环遍历一个每种商品的数量。 和01背包还是如出一辙的。
当然还有那种二进制优化的方法,其实就是把每种物品的数量,打包成一个个独立的包。
和以上在循环遍历上有所不同,因为是分拆为各个包最后可以组成一个完整背包,具体原理我就不做过多解释了,大家了解一下就行,面试的话基本不会考完这个深度了,感兴趣可以自己深入研究一波。
总结:
多重背包在面试中基本不会出现,力扣上也没有对应的题目,大家对多重背包的掌握程度知道它是一种01背包,并能在01背包的基础上写出对应代码就可以了。
至于背包九讲里面还有混合背包,二维费用背包,分组背包等等这些,大家感兴趣可以自己去学习学习,这里也不做介绍了,面试也不会考。
Algorithm and Data Structure Notes
LeetCode BackPack 背包问题
代码随想录
随想录视频
198. 打家劫舍
每个家庭有不同金钱数量,盗窃的家庭不能相邻,求最大盗窃金额。
class Solution:
def rob(self, nums: List[int]) -> int:
if len(nums)==1:
return nums[-1]
dp = [0]*len(nums)
dp[0] = nums[0]
dp[1] = max(nums[0], nums[1])
for i in range(2, len(nums)):
dp[i] = max(dp[i-2]+nums[i], dp[i-1])
return dp[-1]
213. 打家劫舍 II
承接上题,但是房子是环状,即首位相连的。
那么要考虑第一个房子和最后一个房子会相临,只能取其一。
所以,我们把房子变成两种情况:
以上两种情况取最大值后再去他们两个之间的最大值,可以通过上题的解法来解了:
class Solution:
def rob(self, nums: List[int]) -> int:
if len(nums)==1:
return nums[-1]
def roblist(nums):
if len(nums)==1:
return nums[-1]
dp = [0]*len(nums)
dp[0]=nums[0]
dp[1]=max(nums[0], nums[1])
for i in range(2, len(nums)):
dp[i] = max(dp[i-1], dp[i-2]+nums[i])
return dp[-1]
return max(roblist(nums[1:]), roblist(nums[:-1]))
337.打家劫舍 III
题目思路类似第一种类型,但是需要掌握二叉树的遍历。
class Solution:
def rob(self, root: Optional[TreeNode]) -> int:
# dp数组(dp table)以及下标的含义:
# 1. 下标为 0 记录 **不偷该节点** 所得到的的最大金钱
# 2. 下标为 1 记录 **偷该节点** 所得到的的最大金钱
dp = self.traversal(root)
return max(dp)
# 要用后序遍历, 因为要通过递归函数的返回值来做下一步计算
def traversal(self, node):
# 递归终止条件,就是遇到了空节点,那肯定是不偷的
if not node:
return (0, 0)
left = self.traversal(node.left)
right = self.traversal(node.right)
# 不偷当前节点, 偷子节点
val_0 = max(left[0], left[1]) + max(right[0], right[1])
# 偷当前节点, 不偷子节点
val_1 = node.val + left[0] + right[0]
return (val_0, val_1)
像这样的股票问题,使用动态规划,创建两列即两个状态的二维数组,照常初始化,但需要根据题意去得出特定的递推公式,便可完成该题。
121. 买卖股票的最佳时机
贪心算法:
class Solution:
def maxProfit(self, prices: List[int]) -> int:
low = float('inf')
result = 0
# 遍历,每一个值当做最大值,其中可能存在最小值,进行替换
# 右边当最大值,在其左边找最小值
for i in range(len(prices)):
low = min(low, prices[i])
result = max(result, prices[i]-low)
return result
递推公式,两种状态:
要么持有该股票:那么要么之前就买了dp[i-1][0],保持原样,要么刚买-prices[i]
要么不持有该股票:那么之前就没买dp[i-1][1],也保持原样,要么刚卖prices[i]+dp[i-1],取差值。
动态规划:
class Solution:
def maxProfit(self, prices: List[int]) -> int:
# 创建二维数组,两个状态,持有和不持有
length = len(prices)
if length==0:
return 0
dp = [[0, 0] for _ in range(length)]
# 初始化,第一天持有则买入,第一天不持有则仍然为0
dp[0][0] = -prices[0]
dp[0][1] = 0
# 开始遍历
for i in range(1, length):
# 持有股票
dp[i][0] = max(dp[i-1][0], -prices[i])
# 不持有股票
dp[i][1] = max(dp[i-1][1], prices[i]+dp[i-1][0])
return dp[-1][1]
滚动数组:
class Solution:
def maxProfit(self, prices: List[int]) -> int:
length = len(prices)
dp = [[0] * 2 for _ in range(2)] #注意这里只开辟了一个2 * 2大小的二维数组
dp[0][0] = -prices[0]
dp[0][1] = 0
for i in range(1, length):
dp[i % 2][0] = max(dp[(i-1) % 2][0], -prices[i])
dp[i % 2][1] = max(dp[(i-1) % 2][1], prices[i] + dp[(i-1) % 2][0])
return dp[(length-1) % 2][1]
进一步改进:
class Solution:
def maxProfit(self, prices: List[int]) -> int:
length = len(prices)
dp0, dp1 = -prices[0], 0 #注意这里只维护两个常量,因为dp0的更新不受dp1的影响
for i in range(1, length):
dp1 = max(dp1, dp0 + prices[i])
dp0 = max(dp0, -prices[i])
return dp1
122. 买卖股票的最佳时机 II
这里重申一下dp数组的含义:
dp[i][0] 表示第i天持有股票所得现金。
dp[i][1] 表示第i天不持有股票所得最多现金
如果第i天持有股票即dp[i][0], 那么可以由两个状态推出来:
在121. 买卖股票的最佳时机 中,因为股票全程只能买卖一次,所以如果买入股票,那么第i天持有股票即dp[i][0]一定就是 -prices[i]。
而本题,因为一只股票可以买卖多次,所以当第i天买入股票的时候,所持有的现金可能有之前买卖过的利润。那么第i天持有股票即dp[i][0],如果是第i天买入股票,所得现金就是昨天不持有股票的所得现金 减去 今天的股票价格 即:dp[i - 1][1] - prices[i]。
再来看看如果第i天不持有股票即dp[i][1]的情况, 依然可以由两个状态推出来
class Solution:
def maxProfit(self, prices: List[int]) -> int:
# 步骤同上一题
dp = [[0,0] for _ in range(len(prices))]
dp[0][0] = -prices[0]
dp[0][1] = 0
for i in range(1, len(prices)):
#注意这里是和121. 买卖股票的最佳时机唯一不同的地方
dp[i][0] = max(dp[i-1][0], dp[i-1][1]-prices[i])
dp[i][1] = max(dp[i-1][1], dp[i-1][0]+prices[i])
return dp[-1][1]
滚动数组:
class Solution:
def maxProfit(self, prices: List[int]) -> int:
length = len(prices)
dp = [[0] * 2 for _ in range(2)] #注意这里只开辟了一个2 * 2大小的二维数组
dp[0][0] = -prices[0]
dp[0][1] = 0
for i in range(1, length):
dp[i % 2][0] = max(dp[(i-1) % 2][0], dp[(i-1) % 2][1] - prices[i])
dp[i % 2][1] = max(dp[(i-1) % 2][1], dp[(i-1) % 2][0] + prices[i])
return dp[(length-1) % 2][1]
其他解法:
class Solution:
def maxProfit(self, prices: List[int]) -> int:
# 创建dp table
# 含义第N天的最大利润
dp = [0] * (len(prices)+1)
# 初始化,第0天和第一天没有利润,为0,当然这里多余
# dp[0] = dp[1] = 0
# 从第二天开始产生正负利润
for i in range(2, len(prices)+1):
# 当前第N天的利润等于,max(之前的利润,今天的利润+之前的利润)
# 这里取最大值,若今天产生负利润,则不买,只取之前的利润
dp[i] = max(dp[i-1], prices[i-1]-prices[i-2]+dp[i-1])
return dp[-1]
123. 买卖股票的最佳时机 III
class Solution:
def maxProfit(self, prices: List[int]) -> int:
if len(prices) == 0:
return 0
dp = [[0] * 5 for _ in range(len(prices))]
dp[0][1] = -prices[0]
dp[0][3] = -prices[0]
for i in range(1, len(prices)):
dp[i][0] = dp[i-1][0]
dp[i][1] = max(dp[i-1][1], dp[i-1][0] - prices[i])
dp[i][2] = max(dp[i-1][2], dp[i-1][1] + prices[i])
dp[i][3] = max(dp[i-1][3], dp[i-1][2] - prices[i])
dp[i][4] = max(dp[i-1][4], dp[i-1][3] + prices[i])
return dp[-1][4]
官方降低复杂度解法:
class Solution:
def maxProfit(self, prices: List[int]) -> int:
if len(prices) == 0:
return 0
dp = [0] * 5
dp[1] = -prices[0]
dp[3] = -prices[0]
for i in range(1, len(prices)):
dp[1] = max(dp[1], dp[0] - prices[i])
dp[2] = max(dp[2], dp[1] + prices[i])
dp[3] = max(dp[3], dp[2] - prices[i])
dp[4] = max(dp[4], dp[3] + prices[i])
return dp[4]
188. 买卖股票的最佳时机 IV
class Solution:
def maxProfit(self, k: int, prices: List[int]) -> int:
if len(prices) == 0:
return 0
dp = [[0] * (2*k+1) for _ in range(len(prices))]
for j in range(1, 2*k, 2):
dp[0][j] = -prices[0]
for i in range(1, len(prices)):
for j in range(0, 2*k-1, 2):
dp[i][j+1] = max(dp[i-1][j+1], dp[i-1][j] - prices[i])
dp[i][j+2] = max(dp[i-1][j+2], dp[i-1][j+1] + prices[i])
return dp[-1][2*k]
滚动数组:
class Solution:
def maxProfit(self, k: int, prices: List[int]) -> int:
if len(prices) == 0: return 0
dp = [0] * (2*k + 1)
for i in range(1,2*k,2):
dp[i] = -prices[0]
for i in range(1,len(prices)):
for j in range(1,2*k + 1):
if j % 2:
dp[j] = max(dp[j],dp[j-1]-prices[i])
else:
dp[j] = max(dp[j],dp[j-1]+prices[i])
return dp[2*k]
309. 最佳买卖股票时机含冷冻期
class Solution:
def maxProfit(self, prices: List[int]) -> int:
n = len(prices)
if n == 0:
return 0
dp = [[0] * 4 for _ in range(n)]
dp[0][0] = -prices[0] #持股票
for i in range(1, n):
dp[i][0] = max(dp[i-1][0], max(dp[i-1][3], dp[i-1][1]) - prices[i])
dp[i][1] = max(dp[i-1][1], dp[i-1][3])
dp[i][2] = dp[i-1][0] + prices[i]
dp[i][3] = dp[i-1][2]
return max(dp[n-1][3], dp[n-1][1], dp[n-1][2])
714. 买卖股票的最佳时机含手续费
class Solution:
def maxProfit(self, prices: List[int], fee: int) -> int:
n = len(prices)
dp = [[0] * 2 for _ in range(n)]
dp[0][0] = -prices[0] #持股票
for i in range(1, n):
dp[i][0] = max(dp[i-1][0], dp[i-1][1] - prices[i])
dp[i][1] = max(dp[i-1][1], dp[i-1][0] + prices[i] - fee)
return max(dp[-1][0], dp[-1][1])
力扣题目链接
从左开始遍历每个点,再从左到该点进行遍历和比较,比它小则累加,这种加法保证了是有序的。
class Solution:
def lengthOfLIS(self, nums: List[int]) -> int:
# 全部初始化为最小的子序列1,用来存储该位置的最小子序列
dp = [1]*len(nums)
dp[0]=1
for i in range(1, len(nums)):
# 因为是非连续的子序列,所以需要遍历前面所有点
for j in range(i):
if nums[j]<nums[i]:
# 注意这里不是要dp[i] 与 dp[j] + 1进行比较,而是我们要取dp[j] + 1的最大值。
dp[i] = max(dp[i], dp[j]+1)
return max(dp)
力扣题目链接
注意与 1.5.2.2、718. 最长重复子数组 的区分,可以先尝试做718. 最长重复子数组再来尝试这题。
这题要考虑不连续的公共子串,匹配不相等时,该如何处理值:
有三个方向推到dp[i][j],斜对角为匹配相等时,前面匹配的字符+这次匹配到一次也就是1;匹配不相等时,上和左取最大值,取上可能这一行都不匹配,取左可能前面有匹配的而后面不匹配。
class Solution:
def longestCommonSubsequence(self, text1: str, text2: str) -> int:
dp = [[0]*(len(text2)+1) for _ in range(len(text1)+1)]
for i in range(1, len(text1)+1):
for j in range(1, len(text2)+1):
if text1[i-1] == text2[j-1]:
# 匹配相等,把斜上角的值填过来+1,斜上角为各个字符串上一步匹配相等的次数
dp[i][j] = dp[i-1][j-1] + 1
else:
# dp[i][j-1],有过相等但后续匹配不相等时,把左一个值向右填充
# dp[i-1][j],匹配完全不相等,把上一个值向下填充
# dp[i][j-1], text2在该字母上的最大子序列
# dp[i-1][j], text1在该字母上的最大子序列
# 两个字母不相等,
dp[i][j] = max(dp[i-1][j], dp[i][j-1])
return dp[-1][-1]
力扣题目链接
连线是有序的,实际上是求最大公共子序列。
求两个字符串的公共子序列,则需要二元数组。
class Solution:
def maxUncrossedLines(self, nums1: List[int], nums2: List[int]) -> int:
dp = [[0]*(len(nums2)+1) for _ in range(len(nums1)+1)]
for i in range(1, len(nums1)+1):
for j in range(1, len(nums2)+1):
if nums1[i-1]==nums2[j-1]:
dp[i][j] = dp[i-1][j-1]+1
else:
dp[i][j] = max(dp[i-1][j], dp[i][j-1])
print(dp)
return dp[-1][-1]
这一题的本质是与 1.5.1.2、1143. 最长公共子序列 相同,代码与过程分析相同。
力扣题目链接
这道题注意和上个单元的 1.5.1.1、300. 最长上升子序列 的区分,连续和不连续。
class Solution:
def findLengthOfLCIS(self, nums: List[int]) -> int:
dp = [1]*len(nums)
# 这里只用一层循环,因为是连续子串,所以只用比较相邻的值就行
for i in range(1, len(nums)):
if nums[i]>nums[i-1]:
dp[i] = dp[i-1]+1
return max(dp)
或者 增加些内存来减少用时:
class Solution:
def findLengthOfLCIS(self, nums: List[int]) -> int:
dp = [1]*len(nums)
result = 1
for i in range(1, len(nums)):
if nums[i]>nums[i-1]:
dp[i] = dp[i-1]+1
if dp[i]>result:
result = dp[i]
return result
力扣题目链接
class Solution:
def findLength(self, nums1: List[int], nums2: List[int]) -> int:
# 这里创建二维dp table,记录两个字符串的匹配数(注意i与j的顺序)
dp = [[0]*(len(nums2)+1) for _ in range(len(nums1)+1)]
# 取最大值
result = 0
# 两层循环,遍历两个字符串,顺序可以调换
for i in range(1, len(nums1)+1):
for j in range(1, len(nums2)+1):
# 判断当前索引的值是否相等
if nums1[i-1]==nums2[j-1]:
# 等于各个前面一个的值的匹配数+1
dp[i][j] = dp[i-1][j-1]+1
if dp[i][j]>result:
result = dp[i][j]
return result
时间复杂度: O ( n × m ) O(n × m) O(n×m),n 为A长度,m为B长度
空间复杂度: O ( n × m ) O(n × m) O(n×m)
滚动数组:
时间复杂度: O ( n × m ) O(n × m) O(n×m),n 为A长度,m为B长度
空间复杂度: O ( m ) O(m) O(m)
力扣题目链接
class Solution:
def maxSubArray(self, nums: List[int]) -> int:
# 从0到第i个数的最大值
dp = [0]*(len(nums))
# 初始化,第一个
dp[0] = nums[0]
# 从第二个开始遍历
for i in range(1, len(nums)):
# 前一个最大值+该索引的值是否大,大就相加,不大就只保留索引值
dp[i] = max(nums[i], nums[i]+dp[i-1])
return max(dp)
力扣题目链接
思路同上,两个字符串比较,最大公共子序列
class Solution:
def isSubsequence(self, s: str, t: str) -> bool:
dp = [[0]*(len(t)+1) for _ in range(len(s)+1)]
for i in range(1, len(s)+1):
for j in range(1, len(t)+1):
if s[i-1]==t[j-1]:
dp[i][j] = dp[i-1][j-1]+1
else:
# dp[i-1][j]表示不匹配时,是否把上一个结果往下移动,
# 不加会又从0开始,加会少+1一次,总之,结果都会比length少
dp[i][j] = dp[i][j-1]
# 也可以:dp[i][j] = max(dp[i][j-1], dp[i-1][j])
if dp[-1][-1]==len(s):
return True
return False
力扣题目链接
class Solution:
def numDistinct(self, s: str, t: str) -> int:
# dp table 出现的个数
dp = [[0]*(len(s)+1) for _ in range(len(t)+1)]
# t若为空,则在s中都能出现一次,所以第一行初始化为1
for i in range(len(s)+1):
dp[0][i] = 1
# s为空,则不能被t的任何任何字母组成
# 所以 dp[i][0] = 0不变
# i到前面的字母(t) 出现在 j到之前的字母(s) 出现的次数
for i in range(1, len(t)+1):
for j in range(1, len(s)+1):
if t[i-1] == s[j-1]:
# dp[i-1][j-1] 两个字符串s与t,退一个字母匹配的次数
# dp[i][j-1] s字符串 向左退一次匹配的次数,前面有重复的就会累加
dp[i][j] = dp[i-1][j-1]+dp[i][j-1]
else:
dp[i][j] = dp[i][j-1]
return dp[-1][-1]
力扣题目链接
确定递推公式
当word1[i - 1] 与 word2[j - 1]相同的时候
当word1[i - 1] 与 word2[j - 1]不相同的时候
当word1[i - 1] 与 word2[j - 1]相同的时候,dp[i][j] = dp[i - 1][j - 1];
当word1[i - 1] 与 word2[j - 1]不相同的时候,有三种情况:
情况一:删word1[i - 1],最少操作次数为dp[i - 1][j] + 1
情况二:删word2[j - 1],最少操作次数为dp[i][j - 1] + 1
情况三:同时删word1[i - 1]和word2[j - 1],操作的最少次数为dp[i - 1][j - 1] + 2
那最后当然是取最小值,所以当word1[i - 1] 与 word2[j - 1]不相同的时候,递推公式:dp[i][j] = min({dp[i - 1][j - 1] + 2, dp[i - 1][j] + 1, dp[i][j - 1] + 1})
class Solution:
def minDistance(self, word1: str, word2: str) -> int:
# 建表
dp = [[0]*(len(word2)+1) for _ in range(len(word1)+1)]
# 初始化
for i in range(len(word2)+1):
dp[0][i]=i
for i in range(len(word1)+1):
dp[i][0]=i
# 填表
for i in range(1, len(word1)+1):
for j in range(1, len(word2)+1):
if word1[i-1] == word2[j-1]:
dp[i][j] = dp[i-1][j-1]
else:
# word1删除,word2删除,word1和word2都删除一个
dp[i][j] = min(dp[i-1][j]+1, dp[i][j-1]+1, dp[i-1][j-1]+2)
return dp[-1][-1]
力扣题目链接
class Solution:
def minDistance(self, word1: str, word2: str) -> int:
dp = [[0]*(len(word2)+1) for _ in range(len(word1)+1)]
# 初始化,删除为空值需要的步数
for i in range(len(word2)+1):
dp[0][i] = i
for i in range(len(word1)+1):
dp[i][0] = i
# 循环
for i in range(1, len(word1)+1):
for j in range(1, len(word2)+1):
if word1[i-1] == word2[j-1]:
dp[i][j] = dp[i-1][j-1]
else:
# 删除word1,删除word2,替换word1和word2. +1都是一次操作
dp[i][j] = min(dp[i-1][j]+1, dp[i][j-1]+1, dp[i-1][j-1]+1)
return dp[-1][-1]
力扣题目链接
递归公式:
if s[i] == s[j]:
if j-i<=1:
result+=1
dp[i][j] = True
else if dp[i+1][j-1]:
result+=1
dp[i][j] = True
result就是统计回文子串的数量。
这里没有列出当s[i]与s[j]不相等的时候,因为在下面dp[i][j]初始化的时候,就初始为false。
dp数组如何初始化
dp[i][j]不能初始化为true,这就表示全都匹配上了。所以dp[i][j]初始化为false。
确定遍历顺序
首先从递推公式中可以看出,情况三是根据dp[i + 1][j - 1]是否为true,在对dp[i][j]进行赋值true的。
dp[i + 1][j - 1] 在 dp[i][j]的左下角,如图:
如果这矩阵是从上到下,从左到右遍历,那么会用到没有计算过的dp[i + 1][j - 1],也就是根据不确定是不是回文的区间[i+1,j-1],来判断了[i,j]是不是回文,那结果一定是不对的。
所以一定要从下到上,从左到右遍历,这样保证dp[i + 1][j - 1]都是经过计算的。
有的代码实现是优先遍历列,然后遍历行,其实也是一个道理,都是为了保证dp[i + 1][j - 1]都是经过计算的。
注意因为dp[i][j]的定义,所以j一定是大于等于i的,那么在填充dp[i][j]的时候一定是只填充右上半部分。
综上,代码如下:
动态规划:
class Solution:
def countSubstrings(self, s: str) -> int:
# 暴力求解是双循环,所以这里创建仿造,创建二元数组
dp = [[False]*(len(s)) for _ in range(len(s))]
result = 0
# 从后往前遍历
for i in range(len(s)-1, -1, -1):
for j in range(i, len(s)):
# 字符串字母相同
if s[i]==s[j]:
# 自身或者相邻
if j-i<=1:
dp[i][j]=True
result+=1
# 不相邻,相隔大于1
# 该字符串是否为回文=首字母+1,结尾字母-1
# 就是前后都往内部缩一个字母是否为回文
elif dp[i+1][j-1]:
dp[i][j]=True
result+=1
return result
时间复杂度:O(n^2)
空间复杂度:O(n^2)
暴力遍历:
双层循环,每次后移一个字母判断回文
class Solution:
def countSubstrings(self, s: str) -> int:
result = 0
# 暴力遍历
for i in range(1, len(s)+1):
for j in range(i):
if s[j:i] == s[j:i][::-1]:
result+=1
return result
双指针法:
class Solution:
def countSubstrings(self, s: str) -> int:
result = 0
for i in range(len(s)):
# 以i为中心
result += self.extend(s, i, i, len(s))
# 以i+1为中心
result += self.extend(s, i, i+1, len(s))
return result
def extend(self, s, i, j, n):
res = 0
while i>=0 and j<n and s[i]==s[j]:
i-=1
j+=1
res+=1
return res
时间复杂度:O(n^2)
空间复杂度:O(1)
5. 最长回文子串
动态规划:
class Solution:
def longestPalindrome(self, s: str) -> str:
# 用来存储子串字符串
dp = [['']*(len(s)) for _ in range(len(s))]
# 斜对角为1
for i in range(len(s)):
dp[i][i]=s[i]
max_length = ''
# 第一层循环:整体上从后往前开始递归
# 第二层循环:内部从头往后遍历递归
# 相同的已经初始化为1,则从+1下一个开始直到终点
for i in range(len(s)-1, -1, -1):
for j in range(i+1, len(s)):
# 当两个位置的字母相等时,并且(中间为回文)或者(相邻或间隔一个)
if s[i]==s[j] and (dp[i+1][j-1] or (j-i<=2)):
dp[i][j] = s[i]+dp[i+1][j-1]+s[j]
max_length = max(max_length, max(dp[i], key=len), key=len)
# print(dp)
return max_length
单中心点遍历:
选择中心点,一个或者两个,同时向外扩散,边界相等为回文则继续,不相等终止,判断是否为最大回文字符串。
class Solution:
def longestPalindrome(self, s: str) -> str:
max_str = ''
def judge(l,r):
while 0<=l and r<len(s) and s[l]==s[r]:
l-=1
r+=1
nonlocal max_str
max_str = max(max_str, s[l+1:r], key=len)
for i in range(len(s)):
judge(i,i)
judge(i,i+1)
return max_str
相似题型:
HJ32 密码截取 - 牛客网
import sys
for line in sys.stdin:
line = line[:-1]
# dp[i][j]表示从i到j是否为回文
# create dp table
dp = [[0]*len(line) for _ in range(len(line))]
# initial table
for i in range(len(line)):
dp[i][i]=1
max_long = 0
# go through table
# 一头一尾开始遍历,行逆序
for i in range(len(line)-1, -1, -1):
for j in range(i+1, len(line)):
# 字母相等
if line[i] == line[j]:
# 判断中间字符是否相等,或者两个字符相邻,无中间字符
if dp[i+1][j-1] or j-i==1:
# 加上i和j上的两个字符
dp[i][j] = dp[i+1][j-1]+2
# 去掉非连续
# else:
# dp[i][j] = max(dp[i+1][j], dp[i][j-1])
# 每一次起点遍历,保存最大值
if max_long<max(dp[i]):
max_long = max(dp[i])
print(max_long)
这里,可以像上题一样的写法,先不初始化,双循环内的判断条件可以改为,1.奇:i-j=0,2.偶:i-j=1,3.大于2的情况,除开两端的中间为回文,即:
力扣题目链接
这一题与上一题的区别:回文子串是要连续的,回文子序列可不是连续的。
for i in range(len(s)):
dp[i][i]=1
for i in range(len(s)-1, -1, -1):
for j in range(i+1, len(s)):
if s[i] == s[j]:
dp[i][j] = dp[i+1][j-1] + 2
else:
dp[i][j] = max(dp[i+1][j], dp[i][j-1])
class Solution:
def longestPalindromeSubseq(self, s: str) -> int:
# dp[i][j]表示从i到j的最长子序列
dp = [[0]*len(s) for _ in range(len(s))]
# 初始化,对角线为1
for i in range(len(s)):
dp[i][i]=1
# 循环填表,从下到上,从左到右
for i in range(len(s)-1, -1, -1):
for j in range(i+1, len(s)):
if s[i]==s[j]:
dp[i][j] = dp[i+1][j-1] + 2
else:
dp[i][j] = max(dp[i+1][j], dp[i][j-1])
return dp[0][-1]
221. 最大正方形
dp(i,j)=min(dp(i−1,j),dp(i−1,j−1),dp(i,j−1))+1
class Solution:
def maximalSquare(self, matrix: List[List[str]]) -> int:
row, col = len(matrix), len(matrix[0])
dp = [[0]*col for _ in range(row)]
max_width = 0
for i in range(row):
for j in range(col):
if matrix[i][j]=='1':
if i==0 or j==0:
dp[i][j]=1
else:
dp[i][j] = min(dp[i-1][j], dp[i][j-1],dp[i-1][j-1])+1
max_width = max(max(dp[i]), max_width)
return max_width**2
1277. 统计全为 1 的正方形子矩阵
该题与上一题基本一致,但注意求矩阵个数则要把值都加起来。为什么矩阵的值即个数呢?可以自行仿照上题的图验证下。
class Solution:
def countSquares(self, matrix: List[List[int]]) -> int:
row, col = len(matrix), len(matrix[0])
dp = [[0]*col for _ in range(row)]
count = 0
for i in range(row):
for j in range(col):
if matrix[i][j]==1:
if i==0 or j==0:
dp[i][j]=1
else:
dp[i][j] = min(dp[i-1][j], dp[i][j-1],dp[i-1][j-1])+1
count+=dp[i][j]
return count