代码随想录算法训练营day45 | 70. 爬楼梯 （进阶）、322. 零钱兑换、279.完全平方数

70. 爬楼梯 （进阶）

如何和背包问题联系起来？每次可以爬1到m阶楼梯，1阶，2阶，.... m阶就是物品，楼顶就是背包。

物品可以无限次使用，这是一个完全背包问题(决定背包的循环顺序)，并且是求排列问题(决定内外层循环顺序)，和377. 组合总和 Ⅳ几乎是一样的。

n, m = map(int, input().split())
dp = [0] * (n + 1)
dp[0] = 1
for j in range(n + 1):  # 先遍历背包
    for i in range(1,m + 1): # 再遍历物品
        if j >= i:
            dp[j] += dp[j - i]
print(dp[n])

322. 零钱兑换

动态规划五部曲

1. 确定dp数组及下标的含义：凑足总额为j所需钱币的最少个数为dp[j]
2. 确定递推公式：dp[j] = min(dp[j-coins[i]]+1, dp[j])
3. dp数组初始化：dp[0] = 0，其它值都初始化为最大值
4. 确定遍历顺序：完全背包问题，背包遍历是从小到大。本题要求的是最少个数，那么钱币有顺序和没有顺序都可以，都不影响钱币的最小个数，和排列组合都没有关系，因此内外循环没有特定顺序
5. 举例推导dp数组

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):
                if dp[j-coin] != float('inf'):
                    dp[j] = min(dp[j], dp[j-coin]+1)
        if dp[amount] == float('inf'):
            return -1
        return dp[amount]

279.完全平方数

和上一题322. 零钱兑换思想一致，没有差别

class Solution:
    def numSquares(self, n: int) -> int:
        nums = []
        for i in range(n+1):
            if i ** 2 <= n:
                nums.append(i ** 2)
        dp = [float('inf')] * (n+1)
        dp[0] = 0
        for num in nums:
            for j in range(num, n+1):
                if dp[j-num] != float('inf'):
                    dp[j] = min(dp[j], dp[j-num] + 1)
        if dp[n] == float('inf'):
            return -1
        return dp[n]


更简洁的写法
class Solution:
    def numSquares(self, n: int) -> int:
        dp = [float('inf')] * (n + 1)
        dp[0] = 0

        for i in range(1, int(n ** 0.5) + 1):  # 遍历物品
            for j in range(i * i, n + 1):  # 遍历背包
                # 更新凑成数字 j 所需的最少完全平方数数量
                dp[j] = min(dp[j - i * i] + 1, dp[j])

        return dp[n]