给你一个整数 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
思路:转化成从1 4 9 16 …中选择若干个数加起来等于n,并求出最小的个数是多少。经典完全背包问题,每个物品可以使用无限次。直接每个物品体积就是值,权重是1(表示1个)。然后套模板就行。
下面的是优化版本。
class Solution {
public:
int numSquares(int n) {
vector<int> v;
v.push_back(0);
for(int i = 1; i * i <= n; i ++ ) v.push_back(i * i);
vector<int> f(n + 1, 0x3f);
f[0] = 0;
for(int i = 1; i < v.size(); i ++ )
for(int j = v[i]; j <= n; j ++ )
f[j] = min(f[j], f[j - v[i]] + 1);
return f[n];
}
};