1. 问题描述:
给定正整数 n,找到若干个完全平方数(比如 1, 4, 9, 16, ...)使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。给你一个整数 n ,返回和为 n 的完全平方数的最少数量 。完全平方数是一个整数,其值等于另一个整数的平方;换句话说,其值等于一个整数自乘的积。例如,1、4、9 和 16 都是完全平方数,而 3 和 11 不是。
示例 1:
输入:n = 12
输出:3
解释:12 = 4 + 4 + 4
示例 2:
输入:n = 13
输出:2
解释:13 = 4 + 9
提示:
1 <= n <= 10 ^ 4
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/perfect-squares
2. 思路分析:
① 这道题目其实是有很多种的解法,最容易想到的是使用递归来搜索所有可能的组合方案,在所有搜索到的方案中找到组成当前数字n的最少的平方个数。因为每一个数字都是可以使用无限次的所以比较容易想到另外一种是完全背包模型,我们可以将n看成是背包的容量v,平方数字看成是每一种物品的体积,每种物品的价值为1,对于这道题目来说我们需要求解构成n的最少的平方数字的数目,所以dp的含义会发生变化,这里dp[j]的含义是当前体积为j使用的最少的平方数的数目,根据dp的含义进行递推即可,最终dp[n]就是我们要求解的答案,不过提交上去运行需要几秒。
② 此外还可以使用数学的方法解决,需要使用到两个定理:① 拉格朗日四平方和定理 ② 勒让德三平方和定理
拉格朗日四平方和定理:每个正整数均可表示为4个整数的平方和。勒让德三平方和定理:n可以表示成三个整数的充分必要条件是n != (4 ^ a) * (8b + 7),所以答案只可能是,2,3,4中的一种,先判断是否1,如果是1返回1,不是1则判断2...,判断是否由1个平方数字构成其实很简单,只需要判断开方数字的平方是否等于n即可,判断是否是2需要判断n = a * a + b * b,在枚举的时候判断a * 的时候判断b * b == n * n - a * a即可,判断是否是3首先需要将n中所有的因子4整除干净,然后对8取余数判断是否是7,结果不是7说明可以表示为三个整数的平方和,不能那么答案就是4。
3. 代码如下:
完全背包:
import math
class Solution:
def numSquares(self, n: int) -> int:
dp = [100000] * (n + 2)
dp[0] = 0
s = int(math.sqrt(n))
vi = [0]
for i in range(1, s + 1):
vi.append(i * i)
for i in range(1, s + 1):
for j in range(vi[i], n + 1):
# 当前背包容量为j而且使用当前物品vi[i]的最少平方数字的数目
dp[j] = min(dp[j], dp[j - vi[i]] + 1)
return dp[n]
数学:
import math
class Solution:
def check(self, x: int):
n = int(math.sqrt(x))
return n * n == x
def numSquares(self, n: int) -> int:
if self.check(n): return 1
i = 1
while i <= n // i:
if self.check(n - i * i): return 2
i += 1
while n % 4 == 0:
n //= 4
if n % 8 != 7: return 3
return 4