1、数据结构:四平方和定理中仅需要一个整型结构。动态规划中需要长度为n+1的一维数组结构
2、 四平方和定理:
定理内容:任意一个正整数均可表示为4个整数的平方和。
根据定理可知返回结果只能在1,2,3,4中,同时根据定理还可推导出以下结论(自己不会推,看的答案😭)(1)一个数如果含有因子4,那么我们可以把4都去掉,并不影响结果(2)如果一个数除以8余7的话,那么肯定是由4个完全平方数组成。
根据以上内容,首先将所有4的倍数均除以4简化(必须在第一步,否则除8余7会出现问题)。而满足除以8余7的数直接返回4,实现代码如下。
Go
func numSquares(n int) int {
for ; n%4==0; {n/=4}
if n%8==7 { return 4 }
for i:=0; i*i<n; i++{
j:=int(math.Sqrt(float64(n-i*i)))
if i*i+j*j==n {
if i==0 || j==0{
return 1
}else { return 2 }
}
}
return 3
}
3、动态规划
其实看到题目首先想到的就是动态规划,但是我的基础算法掌握好差🤦♀️,动态规划居然有点无从下手。看了一遍答案后决定自己编写试试,以下整理思路。
构建 dp 数组(除dp[0]初始化为0外,其余所有值初始化为INT_MAX),从0~n遍历(因为n为正整数,所以其值至少为1),j从1开始遍历到 i+j*j>n,在这一过程中,对比dp[i+j*j]与dp[i]+1的值,选其中较小的值,为dp[i+j*j]赋值。
Go
func numSquares(n int) int {
var dp []int=make([]int, n+1)
for k, _:=range dp { dp[k]=math.MaxInt32 }
dp[0]=0
for i:=0; i<=n; i++{
for j:=1; i+j*j<=n; j++{
dp[i+j*j]=getMin(dp[i+j*j], dp[i]+1)
}
}
return dp[n]
}
func getMin(a, b int) int {
if a<=b{
return a
}else{ return b }
}
Java
class Solution {
public int numSquares(int n) {
int[] dp=new int[n+1];
Arrays.fill(dp, Integer.MAX_VALUE); //新的数组填充方法
dp[0]=0;
for(int i=0; i<n; i++){
for(int j=1; i+j*j<=n; j++){
dp[i+j*j]=Math.min(dp[i+j*j], dp[i]+1); // Java中的大小比较
}
}
return dp[n];
}
}