算法套路十四——动态规划之背包问题:01背包、完全背包及各种变形

这篇具有很好参考价值的文章主要介绍了算法套路十四——动态规划之背包问题:01背包、完全背包及各种变形。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

算法套路十四——动态规划之背包问题:01背包、完全背包及各种变形

如果对递归、记忆化搜索及动态规划的概念与关系不太理解,可以前往阅读算法套路十三——动态规划DP入门
动态规划 背包问题变种,# 算法套路,算法,动态规划,深度优先
背包DP介绍:https://oi-wiki.org/dp/knapsack/

算法示例一——0/1背包:

0-1背包:有n个物品,第i个物品的体积为w[i],价值为v[i],每个物品至多选一个, 求体积和不超过capacity时的最大价值和,其中i从0开始。

递归+记忆化搜索

  1. 递归函数定义:在0/1背包问题中,递归函数dfs需要2个参数,i和c来表示当前考虑的物品和背包的剩余容量,dfs(i, c)代表的是考虑前i个物品,在背包容量为c的情况下能够获得的最大价值。

  2. 递归方程:在递归过程中,我们需要依次考虑每个物品,然后采用算法套路十——回溯法之子集型回溯进行选或不选的枚举。

    • 如果不选第i个物品,则为dfs(i-1, c),即前i-1个物品在背包容量为c的情况下能够获得的最大价值。如果选第i个物品,则为dfs(i-1, c-w[i]) + v[i],即前i-1个物品在背包容量为c-w[i]的情况下能够获得的最大价值加上第i个物品的价值v[i]。最终,dfs(i, c)的值为以上两种情况的较大值,即转移方程为·dfs(i,c) =max (dfs(i- 1,c),dfs(i- 1,c- w[i]) + v[i])
    • 当背包的剩余容量c小于第i个物品的重量w[i]时,无法选择第i个物品,此时dfs(i, c)的值为dfs(i-1, c)。
    • 故最终递归方程如下所示动态规划 背包问题变种,# 算法套路,算法,动态规划,深度优先
  3. 边界条件:当i小于0,没有物品选择,此时dfs(i, c)的值为0

  4. 所求值:dfs(n-1, capacity),表示考虑前n个物品,在背包容量为capacity的情况下能够获得的最大价值。

def zero_ one_ knapsack(capacity: int, w: List[int], v: List[int]) -> int:
	n = len(w)
	# python的@cache装饰器可以用来寄存函数对已处理参数的结果,以便遇到相同参数可以直接给出答案。
	@cache
	def dfs(i, c):
		if i<0:
			return 0
		if c < w[i]:
			return dfs(i-1, c)
		return max(dfs(i-1, c), dfs(i-1, c-w[i]) + v[i])
	return dfs(n-1, capacity)

二维数组动态规划

  1. .递归函数dfs(i+1,c)——>二维记忆数组dp[i][j]:定义一个二维数组dp,考虑到数组从0开始,为了数组方便我们可以将代码中所有的i,,其中dp[i][j]表示考虑前i个物品,在背包容量为j的情况下能够获得的最大价值

  2. 结合递归方程及边界值,可以得到动态规划的状态转移方程及数组初始化如下所示:
    d p [ i + 1 ] [ j ] = { 0 i = 0 , j > 0 0 i > 0 , j = 0 d p [ i ] [ j ] j < w [ i ] max ⁡ ( d p [ i [ j ] , d p [ i ] [ j − w [ i ] ] + v [ i ] ) j ≥ w [ i ] dp[i+1][j] = \begin{cases} 0 & i=0,j>0\\ 0 & i>0,j=0\\ dp[i][j] & j < w[i]\\ \max(dp[i[j], dp[i][j-w[i]]+v[i]) & j\geq w[i] \end{cases} dp[i+1][j]= 00dp[i][j]max(dp[i[j],dp[i][jw[i]]+v[i])i=0,j>0i>0,j=0j<w[i]jw[i]

  3. 所求值:dp[n ][capacity]

def zero_one_knapsack(capacity: int, w: List[int], v: List[int]) -> int:
    n = len(w)
    #n行capacity + 1列
    dp = [[0] * (capacity + 1) for _ in range(n)]
    for i in range(n):
        for j in range(capacity + 1):
        	#dp[i+1][j] = max(dp[i][j], dp[i][j - w[i]] + v[i]) if j>=w[i] else dp[i+1][j] = dp[i ][j]
            if j < w[i]:
                dp[i+1][j] = dp[i ][j]
            else:
                dp[i+1][j] = max(dp[i][j], dp[i][j - w[i]] + v[i])
    return dp[n][capacity]

空间优化:转换为一维数组

可以将二维数组dp[]转换为一维数组dp[],此时我们需要使用一个双重循环来遍历所有的物品和背包容量,根据0/1背包问题的状态转移方程来更新dp数组的值。
其实该思路就是算法套路十——回溯法之子集型回溯中的思路二——从答案出发的思路,直接枚举所有选择物品i的情况,即枚举在所有不同容积下选择物品i的情况,并用数组记录,因此我们需要对容积从后往前遍历。
在一维数组下的状态转移方程中,f[j] 表示恰好容量为 j 时的最大或最小价值,但是它只与前 i 个物品有关。因为在状态转移时,对于当前正在考虑的第 i 件物品,假如其体积 w[i] 比当前枚举的体积 j 大,那么该物品就无法放入背包中,因此它应该不会对 f[j] 的结果产生影响。在一维数组中,我们并没有明确保存“选取了哪些物品”,而是只保存了阶段 i(前 i 个物品)和决策 j(背包容量)。在状态转移时,我们始终是以 “前i个物品” 作为暗含条件的。

def zero_one_knapsack(capacity: int, w: List[int], v: List[int]) -> int:
    n = len(w)
    dp = [0] * (capacity + 1) 
    # 遍历每个物品
    for i in range(n):
        # 从后往前遍历,更新 dp 数组,容量j仅仅遍历到能选择当前物品的w[i],防止数组越界
        for j in range(capacity, w[i]-1, -1):
        	#枚举所有选择物品i的情况,并记录
            dp[j] = max(dp[j], dp[j-w[i]] + v[i])
    # 返回背包容量下的最大价值
    return dp[capacity]

从后往前遍历的原因是为了避免在更新 dp 数组时重复使用同一个物品,且对容积j进行遍历时我们仅仅遍历到能选择当前物品的w[i]。当我们从后往前遍历时,我们确保在计算 dp[j]
时,dp[j-w[i]] 的值是基于不包含当前物品 i 的子问题的解。这样,我们可以确保每个物品只被使用一次。
如果我们从前往后遍历,那么在计算 dp[j] 时,dp[j-w[i]] 的值可能已经包含了物品 i,这样就可能导致物品 i
被重复使用,从而得到错误的结果。

如物品重量:w = [3, 4, 5] 物品价值:v = [30, 50, 60] 背包容量:capacity = 8

首先,我们看一下遍历物品 0 时从前往后遍历的情况:
动态规划 背包问题变种,# 算法套路,算法,动态规划,深度优先

从前往后遍历时,我们可以看到 dp[6] 的值为 60,这意味着我们在背包容量为 6 的情况下可以获得总价值为 60 的物品。然而,这是不正确的,因为我们实际上重复使用了物品 0(重量为 3,价值为 30)。

现在,我们看一下遍历物品 0 时从后往前遍历的情况:
动态规划 背包问题变种,# 算法套路,算法,动态规划,深度优先
从后往前遍历时,我们可以看到 dp[6] 的值为 30,这是正确的,因为我们只使用了物品 0(重量为 3,价值为 30)一次。
通过这个对比,我们可以看到从后往前遍历可以避免在更新 dp 数组时重复使用同一个物品

循环顺序总结

已知二维状态转移方程,若要转为一维方程,则一般的循环顺序如下所示
若dp[i+1][j]从下图的红色区域转移得到,一般为倒序遍历,即左边与右下

动态规划 背包问题变种,# 算法套路,算法,动态规划,深度优先

若dp[i+1][j]从下图的蓝色区域转移得到,一般为正序遍历,即右边与左下

动态规划 背包问题变种,# 算法套路,算法,动态规划,深度优先

算法示例二——完全背包:

完全背包:N种物品和一个容量为V的背包,每种物品都有无限件可用。第i种物品的体积是v[i],价值是val[i]。求解将哪些物品装入背包可使这些物品的费用总和不超过背包容量,且价值总和最大。
0/1背包和完全背包是两种常见的背包问题,它们的主要区别在于每个物品是否可以重复使用
在0/1背包问题中,每个物品只能使用一次,即要么选择将该物品放入背包中,要么不放入。因此,0/1背包问题也被称为“二进制背包问题”。而在完全背包问题中,每个物品可以使用多次,即可以选择将该物品放入背包中多次。因此,完全背包问题也被称为“多重背包问题”。

递归+记忆化搜索

因为每个物品可以使用多次,所以在解决完全背包问题时,我们需要对每个物品进行循环,以便计算每个物品可以使用多少次,所以在选择当前物品i之后还要继续递归是否选择当前物品i,故完全背包的递归方程动态规划 背包问题变种,# 算法套路,算法,动态规划,深度优先
即:dp(i,j)=max(dp(i-1,j),dp(i,j-v)+val[i]) if c >=w[i] else dfs(i-1, c),这也是与0/1背包的唯一区别,就是在选择物品i之后仍需继续递归物品i

def unbounded_knapsack(capacity: int, w: List[int], v: List[int]) -> int:
	n = len(w)
	# python的@cache装饰器可以用来寄存函数对已处理参数的结果,以便遇到相同参数可以直接给出答案。
	@cache
	def dfs(i, c):
		if i<0:
			return 0
		'''
		if i==0:
			return v[0] if c>=w[0] else 0 
		'''
		if c < w[i]:
			return dfs(i-1, c)
			#与0/1背包的唯一区别,dfs第一个参数仍是i
		return max(dfs(i-1, c), dfs(i, c-w[i]) + v[i])
	return dfs(n-1, capacity)

二维数组动态规划

唯一区别:转移方程如下所示:
d p [ i + 1 ] [ j ] = { 0 i = 0 , j > 0 0 i > 0 , j = 0 d p [ i ] [ j ] j < w [ i ] max ⁡ ( d p [ i [ j ] , d p [ i + 1 ] [ j − w [ i ] ] + v [ i ] ) j ≥ w [ i ] dp[i+1][j] = \begin{cases} 0 & i=0,j>0\\ 0 & i>0,j=0\\ dp[i][j] & j < w[i]\\ \max(dp[i[j], dp[i+1][j-w[i]]+v[i]) & j\geq w[i] \end{cases} dp[i+1][j]= 00dp[i][j]max(dp[i[j],dp[i+1][jw[i]]+v[i])i=0,j>0i>0,j=0j<w[i]jw[i]
dp[i+1][j] = max(dp[i][j], dp[i+1][j-w[i]] + v[i]) if j>=w[i] else dp[i+1][j] = dp[i][j]来更新dp数组的值。最终,dp[n][capacity]的值即为所求的最大价值。

def unbounded_knapsack(capacity: int, w: List[int], v: List[int]) -> int:
    n = len(w)
    dp = [[0] * (capacity + 1) for _ in range(n)]
    for i in range(n):
        for j in range(capacity + 1):
        	#dp[i+1][j] = max(dp[i][j], dp[i+1][j-w[i]] + v[i]) if j>=w[i] else dp[i+1][j] = dp[i][j]
            if j < w[i]:
                dp[i+1][j] = dp[i][j]
            else:
            	#与0/1背包的唯一区别,选择i+1后第一维度仍是i+1
                dp[i+1][j] = max(dp[i][j], dp[i+1][j-w[i]] + v[i])
    return dp[n][capacity]

空间优化:转换为一维数组

与0/1背包类似,完全背包也可以换为一维数组,不过不同的是完全背包一种物品可以选择多次,所以需要从前往后遍历

def unbounded_knapsack(capacity: int, w: List[int], v: List[int]) -> int:
    n = len(w)
    dp = [0] * (capacity + 1)
    for i in range(n):
        for j in range(w[i], capacity + 1):
       		 #与0/1背包的唯一区别:需要从前往后遍历
            dp[j] = max(dp[j], dp[j-w[i]] + v[i])
    return dp[capacity]

背包问题常见变形

  • 至多装capacity,求方案数/最大价值和
  • 恰好装capacity,求方案数/最大/最小价值和
  • 至少装capacity,求方案数/最小价值和

最大、小值问题一般会回归到 方案数 或 数字个数问题, 一般会使用到 max/min 函数约束答案,此时使用inf 初始化来表示极端情况。 每种方式的数组初始及转移方程如何定义可参考下表,且对于每种问题为什么如此设计初始数组及转移方程经过思考相信可以理解,但做题时仍需具体问题具体分析

问题类型 二维初始数组 二维转移方程 最终结果 注意事项 一维初始数组 一维转移方程
求至多容量j的最大价值 dp = [[0]*(capacity+1) for _ in range(n+1)] dp[i+1][j] = max(dp[i][j], dp[i][j-w[i]]+v[i])
if j>=w[i] else
dp[i][j]
dp[n][capacity] 外层循环所有的物品 i,内层循环遍历所有的容量 j f = [0] * (capacity+1) f[j] = max(f[j], f[j-w[i]]+v[i])
求恰好为容量j的最大价值 dp = [[-inf]*(capacity+1) for _ in range(n+1)];dp[0][0]=0 dp[i+1][j] = max(dp[i][j], dp[i][j-w[i]]+v[i])
if j>=w[i] else
dp[i][j]
dp[n][capacity] 初始值 dp[0][0] = 0, dp[0][j] = 负无穷 (j != 0) f = [0] + [-inf] * capacity f[j] = max(f[j], f[j-w[i]]+v[i])
求恰好为容量j的最小价值 dp = [[inf]*(capacity+1) for _ in range(n+1)];dp[0][0]=0 dp[i+1][j] = min(dp[i][j], dp[i][j-w[i]]+v[i])
if j>=w[i] else
dp[i][j]
dp[n][capacity] 初始值 dp[0][0] = 0, dp[0][j] = 正无穷 (j != 0) f = [0] + [inf] * capacity f[j] = min(f[j], f[j-w[i]]+v[i])
求至少容量j的最小价值 dp = [[inf]*(capacity+1) for _ in range(n+1)];dp[0][0]=0 dp[i+1][j] = min(dp[i][j], dp[i][j-w[i]]+v[i])
if j>=w[i] else
dp[i][j]
dp[n][capacity] - f = [0] + [inf] * capacity f[j] = min(f[j], f[j-w[i]]+v[i])
求至多容量j的方案个数 dp = [[1] + [0]*capacity for _ in range(n+1)] dp[i+1][j] += dp[i][j-w[i]]
if j>=w[i] else
dp[i+1][j] =dp[i][j]
dp[n][capacity] 初始值 dp[i][0] = 1 表示每个容量下至少有一种选法即选择不放任何物品 f = [1] + [1] * capacity f[j] += f[j-w[i]]
求恰好为容量j的方案个数 dp = [[1] + [0]*capacity for _ in range(n+1)] dp[i+1][j] += dp[i][j-w[i]]
if j>=w[i] else
dp[i+1][j] =dp[i][j]
dp[n][capacity] 初始值 dp[i][0] = 1 f = [1] + [0] * capacity f[j] += f[j-w[i]]
求至少容量j的方案个数 dp = [[1] + [0]*capacity for _ in range(n+1)] dp[i+1][j] += dp[i][j-w[i]]
if j>=w[i] else
dp[i+1][j] =dp[i][j]
dp[n][capacity] - f = [1] + [0] * capacity f[j] += f[j-w[i]]

算法练习一:LeetCode416. 分割等和子集

给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
动态规划 背包问题变种,# 算法套路,算法,动态规划,深度优先

如表格所示,求恰好容量的方案数,故二维初始数组为dp = [[1] + [0]*capacity for _ in range(n+1)],转移方程为dp[i+1][j] += dp[i][j-w[i]] if j>=w[i] else dp[i+1][j] =dp[i][j] dp[n][capacity],因此我们直接定义zero_one_knapsack_Number_of_schemes()函数来求0/1背包恰好为容量为p的方案数

二维数组动态规划

func canPartition(nums []int) bool {
    s:=sum(nums)
    if s%2!=0{
        return false
    }
    p:=s/2
    if zero_one_knapsack_Number_of_schemes(nums,p)==0{
        return false
    }
    return true
}
//二维动态规划数组求恰好容量为p的0/1背包的方案数,w为体积数组
func zero_one_knapsack_Number_of_schemes(w []int,capacity int)int{
    n:=len(w)
    dp:=make([][]int,n+1)
    for i := range dp {
        dp[i] = make([]int, capacity+1)
    }
    dp[0][0] = 1
    for i:=0;i<n;i++{
        for j:=0;j<=capacity;j++{
            if j<w[i]{
                dp[i+1][j]=dp[i][j]
            }else {
                dp[i+1][j]=dp[i][j]+dp[i][j-w[i]]
            }
        }
    }
    return dp[n][capacity]
}
func sum(nums []int)int{
    sum:=0
    for _,x:=range nums{
        sum+=x
    }
    return sum
}

空间优化为一维数组

一维数组初始化为f = [1] + [0] * capacity,转移方程为f[j] += f[j-w[i]],且定义zero_one_knapsack_Number_of_schemes()函数来求0/1背包恰好为容量为p的方案数

func canPartition(nums []int) bool {
    s:=sum(nums)
    if s%2!=0{
        return false
    }
    p:=s/2
    if one_dimensional_zero_one_knapsack_Number_of_schemes(nums,p)==0{
        return false
    }
    return true
}
//一维动态规划数组求恰好容量为p的0/1背包的方案数
func one_dimensional_zero_one_knapsack_Number_of_schemes(w []int,capacity int)int{
    n:=len(w)
    dp:=make([]int,capacity +1)
    dp[0]=1
    for i:=0;i<n;i++{
    // 从后往前遍历,更新 dp 数组,容量j仅仅遍历到能选择当前物品的w[i],防止数组越界
        for j:=capacity ;j>=w[i];j--{
        //枚举所有选择nums[i]的情况
            dp[j]+=dp[j-w[i]]
        }
    }
    return dp[capacity]
}
func sum(nums []int)int{
    sum:=0
    for _,x:=range nums{
        sum+=x
    }
    return sum
}

算法练习二:LeetCode494. 目标和

给你一个整数数组 nums 和一个整数 target 。
向数组中的每个整数前添加 ‘+’ 或 ‘-’ ,然后串联起所有整数,可以构造一个 表达式 :
例如,nums = [2, 1] ,可以在 2 之前添加 ‘+’ ,在 1 之前添加 ‘-’ ,然后串联起来得到表达式 “+2-1” 。
返回可以通过上述方法构造的、运算结果等于 target 的不同 表达式 的数目。
动态规划 背包问题变种,# 算法套路,算法,动态规划,深度优先

本题看上去与背包问题无关,但实际为0/1背包的变形问题,我们假设数组和为sum,添加+号的数和为p,那么添加-号的和为sum-p,则有p-(sum-p)=target,则2p=target+sum,则题目的要求可以写为选择数组中的数,使其和恰好为(target+sum)/2

二维数组动态规划

如表格所示,求恰好容量的方案数,故二维初始数组为dp = [[1] + [0]*capacity for _ in range(n+1)],转移方程为dp[i+1][j] += dp[i][j-w[i]] if j>=w[i] else dp[i+1][j] =dp[i][j] dp[n][capacity],因此我们直接调用上一题中的zero_one_knapsack_Number_of_schemes()函数来求0/1背包恰好为容量为p的方案数

//p
//s-p
//p-s+p=t
//p=(t+s)/2
//等价于和为p的方案数
func findTargetSumWays(nums []int, target int) int {
    target += sum(nums)
    if target<0||target%2!=0{
        return 0
    }
    p:=target/2
    return zero_one_knapsack_Number_of_schemes(nums,p)
}
//二维动态规划数组求恰好容量为p的0/1背包的方案数,w表示体积数组
func zero_one_knapsack_Number_of_schemes(w []int,capacity int)int{
    n:=len(w)
    dp:=make([][]int,n+1)
    for i := range dp {
        dp[i] = make([]int, capacity +1)
    }
    dp[0][0] = 1
    for i:=0;i<n;i++{
        for j:=0;j<=capacity ;j++{
            if j<w [i]{
                dp[i+1][j]=dp[i][j]
            }else {
                dp[i+1][j]=dp[i][j]+dp[i][j-w[i]]
            }
        }
    }
    return dp[n][capacity]
}
func sum(nums []int)int{
    sum:=0
    for _,x:=range nums{
        sum+=x
    }
    return sum
}

空间优化为一维数组

一维数组初始化为f = [1] + [0] * capacity,转移方程为f[j] += f[j-w[i]],直接调用上一题中one_dimensional_zero_one_knapsack_Number_of_schemes()函数来求0/1背包恰好为容量为p的方案数

func findTargetSumWays(nums []int, target int) int {
    target += sum(nums)
    if target<0||target%2!=0{
        return 0
    }
    p:=target/2
    return one_dimensional_zero_one_knapsack_Number_of_schemes(nums,p)
}
//一维动态规划数组求恰好容量为p的0/1背包的方案数
func one_dimensional_zero_one_knapsack_Number_of_schemes(w []int,capacity int)int{
    n:=len(w)
    dp:=make([]int,capacity +1)
    dp[0]=1
    for i:=0;i<n;i++{
    // 从后往前遍历,更新 dp 数组,容量j仅仅遍历到能选择当前物品的w[i],防止数组越界
        for j:=capacity;j>=w[i];j--{
        //枚举所有选择nums[i]的情况
            dp[j]+=dp[j-w[i]]
        }
    }
    return dp[capacity]
}
func sum(nums []int)int{
    sum:=0
    for _,x:=range nums{
        sum+=x
    }
    return sum
}

算法练习三:LeetCode322. 零钱兑换

给你一个整数数组 coins ,表示不同面额的硬币;以及一个整数 amount ,表示总金额。
计算并返回可以凑成总金额所需的 最少的硬币个数 。如果没有任何一种硬币组合能组成总金额,返回 -1 。
你可以认为每种硬币的数量是无限的。
动态规划 背包问题变种,# 算法套路,算法,动态规划,深度优先

二维数组动态规划

该题中对于不同的硬币可以重复选择,故为完全背包问题的变形。
如表格所示,求恰好金额的最少硬币数,即恰好容量的最小价值,每个物品价值为1,故二维初始数组为dp = [[inf]*(capacity+1) for _ in range(n+1)];dp[0][0]=0,转移方程为dp[i+1][j] = min(dp[i][j], dp[i][j-w[i]]+v[i]) if j>=w[i] else dp[i][j],本题的所有v[i]都表示物品数为1,故定义unbounded_knapsack_Min_items()求恰好容量的的最少物品数。

func coinChange(coins []int, amount int) int {
    
    ans := unbounded_knapsack_Min_items(coins,amount)
    if ans < math.MaxInt/2 {
        return ans
    }
    return -1
}
func unbounded_knapsack_Min_items(nums []int,capacity int)int{
    n := len(nums)
    dp:=make([][]int,n+1)
    for i:=0;i<=n;i++{
        dp[i]=make([]int, capacity+1)
        for j := range dp[0] {
        dp[i][j] = math.MaxInt / 2 // 除 2 是防止下面 + 1 溢出
        }
    }
    dp[0][0]=0
    for i:=0;i<n;i++{
        for j:=0;j<=capacity;j++{
            if j<nums[i]{
                dp[i+1][j]=dp[i][j]
            }else{
                dp[i+1][j]=min(dp[i][j],dp[i+1][j-nums[i]]+1)
            }
        }
    }
    return dp[n][capacity]
}
func min(a, b int) int { if a > b { return b }; return a }

空间优化为一维数组

由表格可知数组初始化为dp = [0] + [inf] * capacity 转移方程为dp[j] = min(dp[j], dp[j-w[i]]+v[i]),本题的所有v[i]都为1,且定义one_dimensional_unbounded_knapsack_Min_items()来求一维数组下恰好容量的最少物品数

func coinChange(coins []int, amount int) int {
    ans := one_dimensional_unbounded_knapsack_Min_items(coins,amount)
    if ans < math.MaxInt/2 {
        return ans
    }
    return -1
}
func one_dimensional_unbounded_knapsack_Min_items(w []int,capacity int)int{
    n := len(w)
    dp:=make([]int,capacity+1)
    for i:=0;i<=capacity;i++{
        dp[i]=math.MaxInt/2
    }
    dp[0]=0

    for i:=0;i<n;i++{
        for j:=w[i];j<=capacity;j++{
            dp[j]=min(dp[j],dp[j-w[i]]+1)
        }
    }
    return dp[capacity]
}
func min(a, b int) int { if a > b { return b }; return a }

算法练习四:LeetCode279. 完全平方数

给你一个整数 n ,返回 和为 n 的完全平方数的最少数量 。1 <= n <= 104
完全平方数 是一个整数,其值等于另一个整数的平方;换句话说,其值等于一个整数自乘的积。例如,1、4、9 和 16 都是完全平方数,而 3 和 11 不是。
动态规划 背包问题变种,# 算法套路,算法,动态规划,深度优先

二维数组动态规划

本题也是对于完全背包的变形,与上题类似,不过其中的物品体积w为1到10000所有的平凡数组成的数组,故直接调用上题的unbounded_knapsack_Min_items()求和为n的最少平方数个数

func numSquares(n int) int {
    w := make([]int, 100)
    for i := 0; i < len(w); i++ {
        w[i] = (i+1) * (i+1)
    }
    return unbounded_knapsack_Min_items(w,n)
}
func unbounded_knapsack_Min_items(nums []int,capacity int)int{
    n := len(nums)
    dp:=make([][]int,n+1)
    for i:=0;i<=n;i++{
        dp[i]=make([]int, capacity+1)
        for j := range dp[0] {
        dp[i][j] = math.MaxInt / 2 // 除 2 是防止下面 + 1 溢出
        }
    }
    dp[0][0]=0
    for i:=0;i<n;i++{
        for j:=0;j<=capacity;j++{
            if j<nums[i]{
                dp[i+1][j]=dp[i][j]
            }else{
                dp[i+1][j]=min(dp[i][j],dp[i+1][j-nums[i]]+1)
            }
        }
    }
    return dp[n][capacity]
}
func min(a, b int) int { if a > b { return b }; return a }

空间优化为一维数组

直接调用上题的one_dimensional_unbounded_knapsack_Min_items()来求一维数组下和为n的最少平方数个数

func numSquares(n int) int {
    w := make([]int, 100)
    for i := 0; i < len(w); i++ {
        w[i] = (i+1) * (i+1)
    }
    return one_dimensional_unbounded_knapsack_Min_items(w,n)
}
func one_dimensional_unbounded_knapsack_Min_items(w []int,capacity int)int{
    n := len(w)
    dp:=make([]int,capacity+1)
    for i:=0;i<=capacity;i++{
        dp[i]=math.MaxInt/2
    }
    dp[0]=0

    for i:=0;i<n;i++{
        for j:=w[i];j<=capacity;j++{
            dp[j]=min(dp[j],dp[j-w[i]]+1)
        }
    }
    return dp[capacity]
}
func min(a, b int) int { if a > b { return b }; return a }

算法练习五:LeetCode518. 零钱兑换 II

给你一个整数数组 coins 表示不同面额的硬币,另给一个整数 amount 表示总金额。
请你计算并返回可以凑成总金额的硬币组合数。如果任何硬币组合都无法凑出总金额,返回 0 。
假设每一种面额的硬币有无限个。
题目数据保证结果符合 32 位带符号整数。动态规划 背包问题变种,# 算法套路,算法,动态规划,深度优先

二维数组动态规划

本题求和为amount的方案数,可以直接修改练习一中的zero_one_knapsack_Number_of_schemes()为unbounded_knapsack_Number_of_schemes()来解决,且由示例可知,0/1背包改为完全背包仅需在选择物品i之后仍继续递归物品i即可

func change(amount int, coins []int) int {
    return unbounded_knapsack_Number_of_schemes(coins,amount)
}

//二维动态规划数组求恰好容量为p的完全背包的方案数,w为体积数组
func unbounded_knapsack_Number_of_schemes(w []int,capacity int)int{
    n:=len(w)
    dp:=make([][]int,n+1)
    for i := range dp {
        dp[i] = make([]int, capacity+1)
    }
    dp[0][0] = 1
    for i:=0;i<n;i++{
        for j:=0;j<=capacity;j++{
            if j<w[i]{
                dp[i+1][j]=dp[i][j]
            }else {
                //唯一区别
                dp[i+1][j]=dp[i][j]+dp[i+1][j-w[i]]
            }
        }
    }
    return dp[n][capacity]
}
func sum(nums []int)int{
    sum:=0
    for _,x:=range nums{
        sum+=x
    }
    return sum
}

空间优化为一维数组

直接修改练习一中的one_dimensional_zero_one_knapsack_Number_of_schemes()为one_dimensional_unbounded_knapsack_Number_of_schemes()来解决,且由示例可知,0/1背包改为完全背包在用一维数组解决时只需要将遍历循序由从后往前修改为从前往后遍历即可文章来源地址https://www.toymoban.com/news/detail-689762.html

func change(amount int, coins []int) int {
    return one_dimensional_unbounded_knapsack_Number_of_schemes(coins,amount)
}
//一维动态规划数组求恰好容量为p的0/1背包的方案数
func one_dimensional_unbounded_knapsack_Number_of_schemes(w []int,capacity int)int{
    n:=len(w)
    dp:=make([]int,capacity +1)
    dp[0]=1
    for i:=0;i<n;i++{
    // 从后往前遍历,更新 dp 数组,容量j仅仅遍历到能选择当前物品的w[i],防止数组越界
        for j:=w[i] ;j<=capacity;j++{
        //枚举所有选择nums[i]的情况
            dp[j]+=dp[j-w[i]]
        }
    }
    return dp[capacity]
}
func sum(nums []int)int{
    sum:=0
    for _,x:=range nums{
        sum+=x
    }
    return sum
}

到了这里,关于算法套路十四——动态规划之背包问题:01背包、完全背包及各种变形的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • 力扣刷题-动态规划算法3:完全背包问题

    问题描述: 1)有N件物品和一个最多能背重量为W的背包。第i件物品的重量是weight[i],得到的价值是value[i] 。 2) 每件物品都有无限个(也就是可以放入背包多次) (比0-1背包多出的条件) 3) 求解将哪些物品装入背包里物品价值总和最大。 求解步骤: 1)首先遍历物品,然

    2023年04月13日
    浏览(39)
  • 代码随想录 Day35 动态规划04 01背包问题和完全背包问题 LeetCode T416 分割等和子集

    说到背包问题大家都会想到使用动规的方式来求解,那么为什么用动规呢, dp数组代表什么呢 ? 初始化是什么 , 遍历方式又是什么 ,这篇文章笔者将详细讲解背包问题的经典例题0-1背包问题和完全背包问题的解题方式,希望能帮助到大家 有人一提到背包问题就只会使用动态规划来

    2024年02月06日
    浏览(43)
  • 动态规划——01背包和完全背包

    目录 01背包模型 题目  dp   滚动数组优化 第一问  第二问  扩展 完全背包 题目  动态规划​编辑  滚动数组优化  关于-1的代码层面优化 💰🪙 背包就是有限制条件的组合问题 有一个背包能容纳的体积是v,现在有n个物品,第i个物品的体积为vi,价值为wi。 (1)求这个背包

    2024年01月20日
    浏览(30)
  • 算法系列--动态规划--背包问题(1)--01背包介绍

    💕\\\"趁着年轻,做一些比较cool的事情\\\"💕 作者:Lvzi 文章主要内容:算法系列–动态规划–背包问题(1)–01背包介绍 大家好,今天为大家带来的是 算法系列--动态规划--背包问题(1)--01背包介绍 背包问题是动态规划中经典的一类问题,经常在笔试面试中出现,是非常 具有区分度 的题

    2024年04月16日
    浏览(30)
  • 算法篇——动态规划 完全和多重背包问题 (js版)

    01 背包 问题和 完全背包 问题的不同点在于,所有的物品只能 使用一次 ,判断  哪些物品  装进背包里 物品价值和 最大;而 完全背包 问题中,所有物品都能 使用n次 ,判断 哪个物品 装 n 个进去 物品价值和 最大。 01 背包的递推公式是: 【当然先遍历物品还是背包的容量

    2024年02月08日
    浏览(33)
  • 算法学习17-动态规划01:背包问题

    提示:以下是本篇文章正文内容: 提示:这里对文章进行总结: 💕💕💕

    2024年04月27日
    浏览(30)
  • 【笔记】动态规划(1)---01背包和完全背包

    集合:选法集合 属性:最优选择 集合的划分 状态表示 集合:所有只考虑 第i个物品前的 且总体积不大于j的所有 选法 。 属性:在所有集和中, 价值最大的选法 。 状态计算 集合的划分:总是在(第 i - 1个) 状态最优时,计算 第 i 个状态。 背包已经最优,故对于任意容积

    2024年02月02日
    浏览(55)
  • 【动态规划】01背包问题——算法设计与分析

    若超市允许顾客使用一个体积大小为13的背包,选择一件或多件商品带走,则如何选择可以使得收益最高? 商品 价格 体积 啤酒 24 10 汽水 2 3 饼干 9 4 面包 10 5 牛奶 9 4 0-1 Knapsack Problem 输入: quad - n n n 个商品组成集合 O O O ,每个商品有属性价格 p i p_i p i ​ 和体积 v i v_i v

    2024年02月04日
    浏览(46)
  • 【Java实现】动态规划算法解决01背包问题

    1、问题描述: 一个旅行者有一个最多能装m公斤的背包,现在有n中物品,每件的重量分别是W1、W2、……、Wn,每件物品的价值分别为C1、C2、……、Cn, 需要将物品放入背包中,要怎么样放才能保证背包中物品的总价值最大? 2、动态规划算法的概述 1)动态规划(Dynamic Progra

    2023年04月09日
    浏览(32)
  • 算法分析与设计——动态规划求解01背包问题

    假设有四个物品,如下图,背包总容量为8,求背包装入哪些物品时累计的价值最多。 我们使用动态规划来解决这个问题,首先使用一个表格来模拟整个算法的过程。 表格中的信息表示 指定情况下能产生的最大价值 。例如, (4, 8)表示在背包容量为8的情况下,前四个物品的最

    2024年02月04日
    浏览(41)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包