【知识迁移】力扣刷题- 完全平方数（Go 实现）

快速通道

279. 完全平方数

279. 完全平方数

给定正整数 n，找到若干个完全平方数（比如 1, 4, 9, 16, …）使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。

给你一个整数 n ，返回和为 n 的完全平方数的 最少数量 。

完全平方数 是一个整数，其值等于另一个整数的平方；换句话说，其值等于一个整数自乘的积。例如，1、4、9 和 16 都是完全平方数，而 3 和
11 不是。

来源：力扣（LeetCode）

解题

这一题其实还是一个背包题，计算的组合数的最小个数，同样是总数，只不过变化的是物品的大小，不过我们知道物品时不能比包大的，如果一开始就初始化一下所有的物品，就回到了熟悉的背包问题了。

func numSquares(n int) int {
    // 背包题
    // 先生成一个完全平方数的队列
    squares := []int{}
    dp := make([]int, n+1)
    for i:=1;i*i <= n; i++ {
        squares = append(squares, i * i)
    }
    // 先初始化dp
    for i:=1;i<=n;i++ {
        dp[i] = i
    }
    // 状态转移方程
    for _,square := range squares {
        for i:=1;i<=n; i++ {
            if i >= square && dp[i - square] + 1 < dp[i] {
                dp[i] = dp[i - square] + 1
            }
            
        }
    }
    return dp[n]
}

优化的话，平方数的列表可以不用生成，嵌入到循环中也是一样的。

func numSquares(n int) int {
    // 背包题
    dp := make([]int, n+1)
    // 先初始化dp
    for i:=1;i<=n;i++ {
        dp[i] = i
    }
    // 状态转移方程
    for j:=1;j*j<=n;j++ {
        square := j*j
        for i:=1;i<=n; i++ {
            if i >= square && dp[i - square] + 1 < dp[i] {
                dp[i] = dp[i - square] + 1
            }
            
        }
    }
    return dp[n]