一、题目描述:279. 完全平方数(中等)
给定正整数 n,找到若干个完全平方数(比如 1, 4, 9, 16, ...)使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。
示例 1:
输入: n = 12
输出: 3
解释: 12 = 4 + 4 + 4.
示例 2:输入: n = 13
输出: 2
解释: 13 = 4 + 9.来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/perfect-squares
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、解题思路
动态规划。
动态数组dp[i]表示数字i最少由多少个平方数相加而成,由于存在1其实最多个数是i个,可以初始化为dp[i] = i。(我本身初始化未-1)
动态方程,当j*j < i时,dp[i] = min(dp[i-j*j] + 1,dp[i]) (1<=j<i)。如果j*j == i那么dp[i] = 1。
三、代码
class Solution {
public:
int numSquares(int n) {
vector<int> dp(n+1,-1);
for(int i = 1; i <= n; i ++)
{
for(int j = 1; j <= n; j ++)
{
int sqr = j*j;
if(sqr > i)
break;
if(sqr == i)
{
dp[i] = 1;
break;
}
if(dp[i] == -1 && dp[i-sqr] != -1)
{
dp[i] = dp[i-sqr] + 1;
}
else if(dp[i] != 1 && dp[i-sqr] != -1)
{
dp[i] = min(dp[i-sqr]+1,dp[i]);
}
}
}
return dp[n];
}
};