1.5 KiB
1.5 KiB
没有问你组合方式,而是问你最小个数
和322 一个套路
// 组合的逻辑
// 版本一
class Solution {
public:
int numSquares(int n) {
vector<int> sum;
for (int i = 1; i * i <= n; i++) {
sum.push_back(i * i);
}
vector<int> dp(n + 1, INT_MAX);
dp[0] = 0;
for (int i = 0; i < sum.size(); i++) {
for (int j = 1; j <= n; j++) {
if (j - sum[i] >= 0 && dp[j - sum[i]] != INT_MAX) {
dp[j] = min(dp[j - sum[i]] + 1, dp[j]);
}
}
}
return dp[n];
}
};
优化一下代码,可以不用预先用sum数组来装i * i,但是版本一更清晰一些,代码如下:
// 版本二
class Solution {
public:
int numSquares(int n) {
vector<int> dp(n + 1, INT_MAX);
dp[0] = 0;
for (int i = 1; i * i <= n; i++) {
for (int j = 1; j <= n; j++) {
if (j - i * i >= 0 && dp[j - i * i ] != INT_MAX) {
dp[j] = min(dp[j - i * i ] + 1, dp[j]);
}
}
}
return dp[n];
}
};
// 排列的逻辑
// 版本三
class Solution {
public:
int numSquares(int n) {
vector<int> dp(n + 1, 0);
for (int i = 0; i <= n; i++) {
dp[i] = i; // 最多也就是i都是1组成的情况
for (int j = 1; j * j <= i; j++) {
dp[i] = min(dp[i - j * j] + 1, dp[i]);
}
}
return dp[n];
}
};