Given a positive integer n, find the least number of perfect square numbers (for example, 1, 4, 9, 16, …) which sum to n.
For example, given n = 12, return 3 because 12 = 4 + 4 + 4; given n = 13, return 2 because 13 = 4 + 9.
思路:动态规划,假设dp[i]表示i的最小完美数个数。
但是,不同的递推公式,时间效率不同。开始,我的想法是
先初始化:
dp[i]=i
然后更改完美数:
dp[i∗i]=1
,
i∗i<=n
接下来递推:
dp[i]=minj<i{dp[i],dp[j]+dp[i−j]}
时间复杂度为:O(n^2),结果超时.代码如下:
class Solution {
public:
int numSquares(int n) {
if(n <= 0) return 0;
int dp[n+1];
for(int i = 0; i <= n; i++) dp[i] = i;
for(int i = 1; i*i <= n; i++) dp[i*i] = 1;
for(int i = 1; i <= n; i++)
{
for(int j = 1; j < i; j++)
{
dp[i] = min(dp[i], dp[j] + dp[i-j]);
}
}
return dp[n];
}
};
后来,看了大神们的解法,真的很巧妙。学习了。
代码如下:
class Solution {
public:
int numSquares(int n) {
if(n<=0) return 0;
int dp[n+1];
fill_n(dp,n+1,0);
for(int i=0;i<=n;i++) {
dp[i]=i;
for(int j=1;j*j<=i;j++) {
dp[i]=min(dp[i],1+dp[i-j*j]);
}
}
return dp[n];
}
};
时间复杂度是O(n^1.5)。