Leetcode 279. 完全平方数
题目
给定正整数 n,找到若干个完全平方数(比如 1, 4, 9, 16, …)使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。
测试样例
示例 1:
输入: n = 12
输出: 3
解释: 12 = 4 + 4 + 4.
示例 2:
输入: n = 13
输出: 2
解释: 13 = 4 + 9.
题解
动态规划
令dp[i]表示组成i的完全平方数的最少个数。我们思考一下转移方程
对于组成i的完全平方数的最少个数肯定是一个数的完全平方数的最少个数+一个完全平方数,也就是说我们需要列举完全平方数,然后进行组合取最小,即dp[i] = min(dp[i],dp[i-j * j]+1),其中需要保证j * j<=i。
我们再考虑一些初始情况
对于0,不可能有完全平方数组成0的;对于1,则只需要1即可。即,dp[0]=0,dp[1]=1。
详细过程见代码
代码
int numSquares(int n) {
vector<int> dp(n+1,INT_MAX);
dp[0] = 0;
dp[1] = 1;
for(int i=2; i<=n; i++){
for(int j=1; j*j<=i; j++){
dp[i] = min(dp[i],dp[i-j*j]+1);
}
}
return dp[n];
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/perfect-squares
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。