主要总结了完全背包问题
对应leetcode题目为518.377.279.70.139
代码地址https://github.com/Ltd5552/Algorithm/tree/master/8-DP/Bag
理论
有N件物品和一个最多能背重量为W的背包。第i件物品的重量是weight[i],得到的价值是value[i] 。每件物品都有无限个(也就是可以放入背包多次),求解将哪些物品装入背包里物品价值总和最大。
之前01背包提到了必须倒序遍历,否则会导致同一物品被多次加入,完全背包问题正好就可以多次加入同一物品,所以只需要在01背包的基础上更改遍历顺序为顺序遍历即可。
同时,由于递推方向是从左往右,物品可以多次放进去,所以不论先遍历物品还是先遍历背包都可以,因为即使先遍历背包也能够保证之前的计算过了,不会出现只有一个物品的情况。
// ComBagSolutionI 先遍历物品再遍历背包
func ComBagSolutionI(weight, value []int, bagWeight int) int {
dp := make([]int, bagWeight+1)
for i := range weight {
//正向遍历,可以重复装入某一个物品
for j := weight[i]; j <= bagWeight; j++ {
//一维dp是靠前面的推导出来的
dp[j] = utils.Max(dp[j], dp[j-weight[i]]+value[i])
}
}
return dp[bagWeight]
}
// ComBagSolutionII 先遍历背包容量再遍历物品
func ComBagSolutionII(weight, value []int, bagWeight int) int {
dp := make([]int, bagWeight+1)
for j := 0; j <= bagWeight; j++ {
//正向遍历,可以重复装入某一个物品
for i := range weight {
if j >= weight[i] {
//一维dp是靠前面的推导出来的
dp[j] = utils.Max(dp[j], dp[j-weight[i]]+value[i])
}
}
}
return dp[bagWeight]
}
518.零钱兑换II
分析
与开篇的纯完全背包问题不同,纯完全背包是求装包的最大价值,而这里是装满的物品组合个数,和之前的目标和差不多。
组合不强调元素之间的顺序,排列强调元素之间的顺序。
动规:
-
定义dp[j]为凑成金额j的货币组合个数;
-
dp[j]就是全部dp[j-coins[i]]情况相加,所以递推公式就是
dp[j] += dp[j-coins[i]]
,这里也和目标和一样; -
如何初始化?还是dp[0] = 1,这是后面所有的基础,如果为0的话全都为0了,其他的默认0即可,这样才不会影响正确的结果,本身开始也没啥意义。
-
遍历顺序
完全背包问题的遍历顺序依旧非常关键,对于纯完全背包问题而言因为是求最大价值,不管怎么排列组合都行,结果没影响;但这里是求装满的方案数,这里的方案数明确要求是顺序无关的,即组合数。
如果先遍历背包再遍历物品,那么对背包容量的每一个值都会经过全部物品,这样针对的是前面的所有可能的取值,比如计算i=4的时候,假设存在{1, 3}那么就会包含
dp[4] += dp[4-1]
和dp[4]+=dp[4-3]
两种排列(但只是一种组合),这样结果就是排列数;如果先遍历物品再遍历背包,那么对于每一个物品的遍历只会“占据一行”,这一次遍历完了后面就不会再出现了,所以上面的例子也只会出现
dp[4] += dp[4-1]
的情况(1在5前面),而不会说遍历完5了又来遍历一次1,他们的相对顺序一定是绝对的(1在5前面就在5前面,不会跑到5后面去,这样自然就不会出现{5, 1}的情况了),这样结果就是组合数。func change(amount int, coins []int) int { dp := make([]int, amount+1) dp[0] = 1 for i := range coins { for j := coins[i]; j <= amount; j++ { dp[j] += dp[j-coins[i]] } } return dp[amount] }
对比的代码在这里可以查看
377.组合总和IV
分析
经过上一题的分析这道题显然就是求排列数了,目标target即背包容量,nums即元素的数组,价值和重量同样都是对应的数值。
具体区别就只有遍历顺序不同。
func combinationSum4(nums []int, target int) int {
dp := make([]int, target+1)
dp[0] = 1
for i := 1; i <= target; i++ {
for j := range nums {
if i >= nums[j] {
dp[i] += dp[i-nums[j]]
}
}
}
return dp[target]
}
279.完全平方数
动规
求和为n的完全平方数最小个数,那么就是背包容量为n,物品是1/4/9/16…的完全背包问题了。
定义dp[i]为当容量为i时满足装满条件下完全平方数的最少个数,递推式dp[i] = min(dp[i], dp[i-nums[j]]+1)
这里的初始化值得注意,因为是求最少个数,那么就得初始化为一个较大的值在后续才能去更小的,所以这里初始化为math.MaxInt
求最小数,与排列组合没有关系,所以先遍历物品或先遍历背包都可以。
func numSquares(n int) int {
dp := make([]int, n+1)
for i := 1; i <= n; i++ {
dp[i] = math.MaxInt
}
for i := 1; i*i <= n; i++ {
for j := i * i; j <= n; j++ {
dp[j] = utils.Min(dp[j], dp[j-i*i]+1)
}
}
return dp[n]
}
数学
依据:
由拉格朗日的四平方定理可得,每个自然数都可以表示为四个整数平方之和。
分析:
四平方和定理证明了任意一个正整数都可以被表示为至多四个正整数的平方和。这给出了本题的答案的上界即4。
四平方和定理可以推出三平方和推论:当且仅当n!=4^k*(8*m+7)
时n可以被表示为至多三个正整数的平方和。
所以n=4^k*(8*m+7)
时只能被表示为四个正整数的平方和。此时我们可以直接返回 4。当n!=4^k*(8*m+7)
时需要判断 n 到底可以分解成几个完全平方数之和。答案肯定是 1,2,3 中的一个。
题目要求我们求最小的,所以从1开始一个个判断是否满足。如果答案为1,代表n为完全平方数,这很好判断。如果答案为 2,代表n=a^2+b^2,枚举1->根号n即可,如果还不满足那么就是3个。
// 执行耗时:0 ms,击败了100.00% 的Go用户
// 内存消耗:1.9 MB,击败了98.97% 的Go用户
func numSquaresFast(n int) int {
if isPerfectSquare(n) {
return 1
}
if checkAnswer4(n) {
return 4
}
for i := 1; i*i <= n; i++ {
j := n - i*i
if isPerfectSquare(j) {
return 2
}
}
return 3
}
// 判断是否为完全平方数
func isPerfectSquare(n int) bool {
sq := int(math.Floor(math.Sqrt(float64(n))))
return sq*sq == n
}
// 判断是否能表示为 4^k*(8m+7)
func checkAnswer4(x int) bool {
for x%4 == 0 {
x /= 4
}
return x%8 == 7
}
正如其介绍的那样,只有beats 100%的题解才会被放到这上面(当然仅限于Golang),所以如果你想知道的最优解或许可以参考这本书,它也仍在不断更新。
70.爬楼梯
分析
完全背包做法就是以步数作为物品{1, 2}来装满背包容量为n的方法有多少种,而且是求的排列数,顺序有关系
若题目将步数扩展为一次可以爬1、2、3、4…那么使用完全背包问题就很简单了,每次的阶数就是物品的重量
// 执行耗时:0 ms,击败了100.00% 的Go用户
// 内存消耗:1.9 MB,击败了21.29% 的Go用户
func climbStairsII(n int) int {
steps := []int{1, 2}
dp := make([]int, n+1)
dp[0] = 1
for i := 0; i <= n; i++ {
for j := range steps {
if i >= steps[j] {
dp[i] += dp[i-steps[j]]
}
}
}
return dp[n]
}
139.单词拆分
分析
将单词看作是物品,字符串看为背包,单词能否组成字符串就是问物品能否将背包装满,同时单词可以重复使用那么就是完全背包问题;再同时因为单词的顺序会改变字符串,这里强调了顺序是有意义的,所以是求排列数。
综上,可以转换为一道求完全背包问题的排列数。
定义dp[i]为当字符串长度为i的时候可以拆分为一个或多个字典中的单词。
递推公式:如果dp[j]为true,s[j, i]也在字典里面那么dp[i]就为true,这里的s[j, i]就是一个单词。
求排列数,先遍历背包再遍历物品。
func wordBreak(s string, wordDict []string) bool {
//创建一个集合用来判断是否存在,单纯的set不方便判断是否存在
wordDictSet := make(map[string]bool)
for _, w := range wordDict {
wordDictSet[w] = true
}
dp := make([]bool, len(s)+1)
dp[0] = true
//排列数,外层遍历背包,内层遍历物品
for i := 1; i <= len(s); i++ {
for j := 0; j < i; j++ {
dp[i] = dp[j] && wordDictSet[s[j:i]]
//找到一组就不用再找了
if dp[i] {
break
}
}
}
return dp[len(s)]
}