给定正整数 n,找到若干个完全平方数(比如 1, 4, 9, 16, …)使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。
示例 1:
输入: n = 12
输出: 3
解释: 12 = 4 + 4 + 4.
示例 2:
输入: n = 13
输出: 2
解释: 13 = 4 + 9.
方法一:使用回溯法(深度优先)。从n的平方根到1进行搜索,直到当前的n为零。
class Solution {
private:
int result;
//在steps之后将n转变成reminaNum后,继续搜索
void dfs(int reminaNum, int steps) {
if (reminaNum == 0) {//如果正好匹配完成
result = steps;//更新结果
return;
}
else if (result > steps) {//剪枝(只有当前的步数少于当前已经搜索到的最小结果才有继续搜索的必要
//sqrt(reminaNum)将reminaNum开方
for (int num = sqrt(reminaNum); num > 0; --num) {
//因为是steps最小,所以从大到小进行搜索
dfs(reminaNum - num * num, steps + 1);
}
}
}
public:
int numSquares(int n) {
result = INT_MAX;
dfs(n, 0);//开始搜素
return result;
}
};