题目:
给定正整数 n,找到若干个完全平方数(比如 1, 4, 9, 16, …)使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。
给你一个整数 n ,返回和为 n 的完全平方数的 最少数量 。
完全平方数 是一个整数,其值等于另一个整数的平方;换句话说,其值等于一个整数自乘的积。例如,1、4、9 和 16 都是完全平方数,而 3 和 11 不是。
示例:
输入:n = 12
输出:3
解释:12 = 4 + 4 + 4
思路:
动态规划
完全背包问题,对于本题就是有一个一定容量的背包,然后有一个集合,这个集合里有各种重量的物品,把这些物品放进背包中,使得背包到达它的容量,并且背包里放的物品数量最少。
背包容量就是n,各种重量的物品就是平方数,这些平方数肯定要小于或者等于背包容量,不然也不能往里放。
- 先建立一个数组,里面放着背包里可以放的物品。
- 建立一个长度为n+1的动态数组,其中dp[i]表示和为i的完全平方数的最少数量。初始化里面是最大数量,题目中说最大容量是10的四次方,可选择的最小重量是1,所以最多数量就是10的四次方。
- 定义dp[0] = 0,要找的是dp[n]的最小数,怎么找呢,我们对于背包里面可以放的物品,从小到大依次往里放,首先对于1这个数,遍历动态数组,肯定要从索引等于1的位置开始,因为索引小于1的位置,也就是背包容量小于1的话,根本放不进,这一趟是只能往里放1的情况下的物品数量,然后遍历下一个物品4,从索引4开始更新物品数量,取最小的物品数量,直到遍历完可放的全部数,最后返回dp[n],即和为n的完全平方数的最小数量。
class Solution(object):
def numSquares(self, n):
"""
:type n: int
:rtype: int
"""
nums = [i ** 2 for i in range(1,int(n**0.5)+1)]
dp = [10**4] * (n+1)
dp[0] = 0
for num in nums:
for j in range(num,n+1):
dp[j] = min(dp[j],dp[j-num]+1)
return dp[n]