题目描述(中等)
给定正整数 n,找到若干个完全平方数(比如 1, 4, 9, 16, …)使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。
给你一个整数 n ,返回和为 n 的完全平方数的 最少数量 。
完全平方数 是一个整数,其值等于另一个整数的平方;换句话说,其值等于一个整数自乘的积。例如,1、4、9 和 16 都是完全平方数,而 3 和 11 不是。
示例 1:
输入:n = 12
输出:3
解释:12 = 4 + 4 + 4
示例 2:
输入:n = 13
输出:2
解释:13 = 4 + 9
提示:
1 <= n <= 104
思路
直接朴素的贪心会出错,就是由小于n的最大的完全平方数来组成n
最直接的例子就是12 方案应该是4+4+4而不是9+1+1+1,也就是不能贪心的去选9
所以想到动态规划,dp[i]是满足和为i的答案
对于i,可以由j 和 i-j,两部分组成
dp[i] = min(dp[i],所有可能的组合dp[j] + dp[i-j])
因为加上了对i的分割,这样复杂度O(n^2),TLE了
进行优化,在对i进行分隔的过程中,用一点贪心的思想
每次划分的过程中,步长不再是1,而是1 4 9 16…这样的完全平方数
步子迈大点,因为步子迈小了结果肯定只会大不会小,因为用了太多的1
这样复杂度就降到O(n sqrt(n))
另外官方题解的数学法真是长见识了,虽然在推导的过程中感觉到了答案应该就那几个数,没想到前人已经总结出来了。官方题解 ,四平方和定理
代码
class Solution {
public:
int numSquares(int n) {
vector<int> dp(10005,6666666);
for(int i = 1; i <= 100 ; i++){
dp[i*i] = 1;
}
for(int i = 2; i <= n; i++){
for(int j = i-1,k = 1;j>=i/2;j = i-k*k) {
dp[i] = min(dp[i],dp[j] + dp[i-j]);
k++;
}
}
return dp[n];
}
};