leetcode hot 100-279. 完全平方数

最新推荐文章于 2024-03-13 23:40:52 发布

Lucky小黄人

最新推荐文章于 2024-03-13 23:40:52 发布

阅读量172

点赞数

文章标签：算法 leetcode 动态规划数据结构快速排序

本文链接：https://blog.csdn.net/qq_41767116/article/details/113734374

版权

279. 完全平方数

题目描述

给定正整数 n，找到若干个完全平方数(比如 1, 4, 9, 16, ...)使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。

示例 1:

输入: n = 12
输出: 3 
解释: 12 = 4 + 4 + 4.

示例 2:

输入: n = 13
输出: 2
解释: 13 = 4 + 9.

思路一：暴力 dfs (时间超时)

先给n开平方，确定上界，假定根号n等于k

递归，从1-k中选出多个值，每个值可选可不选，而且还能重复选，记录选中的次数

当前和如果刚好等于n, 则更新cnt

 1 class Solution {
 2     public int numSquares(int n) {
 3         if(n <= 0){
 4             return 0;
 5         }
 6 
 7         // 先给n开平方，确定上界，假定 根号n等于k
 8         int sqrt = (int)Math.sqrt(n);
 9         traceBack(n, 0, sqrt, 0);
10         return cnt;
11     }
12 
13     public int cnt = 1 << 30;
14 
15     // 递归，从1-k中选出多个值，每个值可选可不选，而且还能重复选，记录选中的次数
16     public void traceBack(int n, int nowSum, int index, int count){
17         // 越界，或者如果大于n, 直接返回
18         if(index <= 0 || n < nowSum){
19             return;
20         }
21         // 边界如果刚好等于n, 则更新cnt
22         if(n == nowSum){
23             cnt = Math.min(cnt, count);
24         }
25 
26         // 对于每个数可选也不可选
27         // 选
28         if(nowSum + index * index <= n){
29             traceBack(n, nowSum + index * index, index, count + 1);
30         }
31 
32         // 不选
33         traceBack(n, nowSum, index-1, count);
34     }
35 }

通过了287个测试用例，然后就超时了

复杂度分析：

时间复杂度：对小于√n 的所有数都进行枚举，而且每个数还不止枚举一遍，所以这个复杂度就不太好算了，我觉得是 2ⁿ√n

空间复杂度：递归栈的深度，我觉得是√n

思路二：动态规划

参考：https://leetcode-cn.com/problems/perfect-squares/solution/hua-jie-suan-fa-279-wan-quan-ping-fang-shu-by-guan/

dp[i]表示组成i的最少完全平方的个数

i 和 (i - j * j)的最终结果就相差一个完全平方数j*j, 所以dp[i] 和 dp[i-j*j]就相差1

dp[i] = Math.min(dp[i], dp[i - j*j] + 1); dp[0] = 0;

边界值dp[i] = i, 因为组成i的最少完全平方的个数最大为i, 即全是由1组成的

 1 class Solution {
 2     public int numSquares(int n) {
 3         if(n <= 0){
 4             return 0;
 5         }
 6 
 7         int[] dp = new int[n+1];
 8         dp[0] = 0;
 9         for(int i = 1; i <= n; i++){
10             dp[i] = i;
11             for(int j = 0; i - j*j >= 0; j++){
12                 dp[i] = Math.min(dp[i], dp[i - j*j] + 1);
13             }
14         }
15         return dp[n];
16     }
17 }

leetcode 执行用时：42 ms > 64.61%, 内存消耗：37.5 MB > 94.62%