给定正整数 n,找到若干个完全平方数(比如 1, 4, 9, 16, …)使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。
示例 1:
输入: n = 12
输出: 3
解释: 12 = 4 + 4 + 4.
示例 2:
输入: n = 13
输出: 2
解释: 13 = 4 + 9.
这一题我觉得和零钱兑换的那一题很像,可以看我之前的博客
LeetCode Hot-322零钱兑换
这个和上面的零钱兑换有一个不同的是,零钱兑换的里面给了具体的零钱的面值,而这个需要你自己去确定,也就是平方数
动态规划
具体思路是:
获取到指定的数的平方根,也就是类似于上面的题中的最大的零钱
初始化dp数组,初值是n+1,因为要求最小值
状态转移方程是
dp[i] = min(dp[i], dp[i - j * j] + 1)
代码
class Solution:
def numSquares(self, n: int) -> int:
max_item = int(math.sqrt(n))
dp = [n + 1 for _ in range(n + 1)]
dp[0] = 0
for i in range(1, n + 1):
for j in range(1, max_item + 1):
if i - j * j >= 0:
dp[i] = min(dp[i], dp[i - j * j] + 1)
return dp[n]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/perfect-squares
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。