
思路:
还是比较好想的,gi定义为和为 i 的完全平方数的最少数量。那么递推关系式是gi=min(gi-1,gi-4,gi-9,...)+1,数组初始化是g0=0,g1=1。注意这里要对g0初始化,(举个例子)因为在遍历到g4时,g4=min(g4-1,g4-4)+1。
代码:
C++:
cpp
class Solution {
public:
int numSquares(int n) {
vector<int> g(n+1,0x3f3f3f3f);
g[0]=0,g[1]=1;
for(int i=2;i<=n;i++){
for(int j=1;i-j*j>=0;j++){
int temp=j*j;
g[i]=min(g[i],g[i-temp]+1);
}
}
return g[n];
}
};
Python:
python
class Solution:
def numSquares(self, n: int) -> int:
g=[0x3f3f3f3f]*(n+1)
g[0]=0
g[1]=1
for i in range(2,n+1):
j=1
while i-j*j>=0:
temp=j*j
g[i]=min(g[i],g[i-temp]+1)
j+=1
return g[n]