给定正整数 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 - j*j + j*j = i,dp[i-j*j] + 1是可能的答案
dp[i] = min(dp[i-j*j]) + 1, j*j<=i
class Solution {
public:
int numSquares(int n) {
int *dp = new int[n + 1]();
for(int i = 1; i <= n; i++)
{
int minn = INT_MAX;
for(int j = 1; j * j <= i; j++)
{
minn = min(minn, dp[i - j * j]);
}
dp[i] = minn + 1;
}
return dp[n];
}
};
也可用BFS做,假设正整数是图上一个点,如果两个正整数的差是完全平方数,那么这两个数有一条边相连,计算从0到n的图上最短路径就是此题的答案
class Solution {
private:
//生成小于n的平方数
vector<int> fun(int n)
{
vector<int> square;
for(int i = 1; i * i <= n; i++)
square.push_back(i * i);
return square;
}
public:
int numSquares(int n)
{
vector<int> picture = fun(n);
bool *vis = new bool[n + 1]();
queue<int> q;
q.push(n);
vis[n] = true;
int res = 1;
while(!q.empty())
{
int size = q.size();
//遍历这一层的所有节点
for(int i = 0; i < size; i++)
{
int cur = q.front();
q.pop();
//找下一层节点
for(int num : picture)
{
//如果两个数的差为平方数,有边
int next = cur - num;
if(next < 0) break;
if(next == 0) return res;
//如果在上面已经访问过
if(vis[next]) continue;
//如果没有访问过并且有边
vis[next] = true;
q.push(next);
}
}
res++;
}
return -1;
}
};