1. 题目
给定正整数 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
Related Topics 广度优先搜索 数学 动态规划
👍 889 👎 0
2. 题解
2.1 解法1: 暴力DFS(超时)
相当于暴力解法, 考虑所有的分解方案,找出最小的解
比如: n = 12
先把 n 减去一个平方数,然后求剩下的数分解成平方数和所需的最小个数
class Solution {
public int numSquares(int n) {
if (n == 0) {
return 0;
}
int count = Integer.MAX_VALUE;
for (int i = 1; i * i <= n; i++) {
count = Math.min(count, numSquares(n - i * i) + 1);
}
return count;
}
}
优化, 由于中间会有大量的重复计算, 所以使用一个map 保存中间值
class Solution {
public int numSquares(int n) {
return dfs(n,new HashMap<>());
}
public int dfs(int n, Map<Integer, Integer> map) {
if (map.containsKey(n)) {
return map.get(n);
}
if (n == 0) {
return 0;
}
int count = Integer.MAX_VALUE;
for (int i = 1; i * i <= n; i++) {
count = Math.min(count, dfs(n - i * i, map) + 1);
}
map.put(n, count);
return count;
}
}
2. 解法 1: 动态规划
- 状态数组: dp[i] , 代表和为 i 的完全平方数的 最少数量
- 递推方程: dp[i] = min(dp[i], dp[i-jj]+1),其中, j是平方数, j=1~k,其中kk要保证 <= i
意思就是:完全平方数和为i的 最小个数 等于 当前完全平方数和为i的 最大个数
与 (完全平方数和为 i - j * j 的 最小个数 + 完全平方数和为 j * j的 最小个数)
可以看到 dp[j*j] 是等于1的 - 初始化: 初始状态dp[i]均取最大值i,即 1+1+…+1,i个1; dp[0] = 0
class Solution {
public int numSquares(int n) {
int[] dp = new int[n + 1];
for (int i = 0; i < dp.length; i++) {
dp[i] = i;
}
for (int i = 1; i <= n; i++) {
for (int j = 1; i - j * j >= 0; j++) {
dp[i] = Math.min(dp[i], dp[i - j * j] + 1);
}
}
return dp[n];
}
}
参考:
详细通俗的思路分析,多解法
[]