Given a positive integer n, find the least number of perfect square numbers (for example, ) which sum to n. Example 1: Example 2: | 给定正整数 n,找到若干个完全平方数(比如 1, 4, 9, 16, ...)使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。 示例 1: 输入: n = 12 输出: 3 解释: 12 = 4 + 4 + 4. 示例 2: 输入: n = 13 输出: 2 解释: 13 = 4 + 9. 在真实的面试中遇到过这道题? |
思路:建一个n+1长度的dp数组,将第一个值初始化为0,其余值都初始化为INT_MAX,i从0循环到n,j从1循环到i+j*j<=n的位置,然后每次更新dp[i+j*j]的值,动态更新dp数组,dp[i]表示正整数i至少能由多少个完全平方数组成,那也就是求n,最终返回dp[n]即可。
class Solution {
public:
int numSquares(int n) {
vector<int> dp(n+1 ,INT_MAX);
dp[0]= 0;
for(int i=0;i<=n;++i)
{
for(int j=1;i+j*j<=n;++j)
dp[i+j*j]=min(dp[i+j*j],dp[i]+1);
}
return dp.back();
}
};