力扣 完全平方数

动态规划,找到前几个状态做更新。

题目

从题可看出又是一道dp,只要找到一个最大的平方数,然后往回退到上个状态,然后再用回退的状态加回去这个平方数即加上这一种。注意这里的所含平方数并不是随着数字变大而变大的,因此还要加多一层循环做遍历的维护,目的是找到的平方数少。

java 复制代码
class Solution {
    public int numSquares(int n) {
        int[] f = new int[n + 1];
        for (int i = 1; i <= n; i++) {
             f[i] = Integer.MAX_VALUE;
            for (int j = 1; j * j <= i; j++) {
                f[i] = Math.min(f[i], f[i - j * j]+1);//通过减去一个平方数对前面已经遍历的f[i]进行筛选
            }
        }
        return f[n];
    }
}

这里对f[i]做了频繁更新,实际只需要在后面更新一次即可,在做比较时可以用一个临时变量去存,这样就可以优化一下维护状态的数组了。

java 复制代码
class Solution {
    public int numSquares(int n) {
        int[] f = new int[n + 1];
        for (int i = 1; i <= n; i++) {
            int minn = Integer.MAX_VALUE;
            for (int j = 1; j * j <= i; j++) {
                minn = Math.min(minn, f[i - j * j]);//通过减去一个平方数对前面已经遍历的f[i]进行筛选
            }
            f[i] = minn + 1;//更新当前数时加回去j*j这种情况
        }
        return f[n];
    }
}

然后也可以换一下内外层循环,先去生成所有完全平方数,然后做更新。

时间复杂度:O(n√n),空间复杂度:O(n)。

java 复制代码
class Solution {
    public int numSquares(int n) {
       
        int[] f = new int[n + 1]; 
        Arrays.fill(f, Integer.MAX_VALUE); 
        f[0] = 0;
        
        for (int i = 1; i * i <= n; i++) {
            for (int j = i * i; j <= n; j++) {

                f[j] = Math.min(f[j], f[j - i * i] + 1);
            }
        }
        
        return f[n];
    }
}

在做dp时,学会找到状态间的关系,也要注意维护状态的数组优化。

相关推荐
有一个好名字5 分钟前
设计模式-单例模式
java·单例模式·设计模式
业精于勤的牙7 分钟前
浅谈:算法中的斐波那契数(六)
人工智能·算法
小孟的CDN7 分钟前
使用pytorch进行batch_size分批训练,并使用adam+lbfgs算法——波士顿房价预测
pytorch·算法·batch·代码·adam+lbfgs
2301_797312268 分钟前
学习Java26天
java·开发语言
cike_y9 分钟前
JSP原理详解
java·开发语言·jsp
invicinble11 分钟前
关于springboot引入traceid来保障可观测型
java·spring boot·后端
仰泳的熊猫12 分钟前
1037 Magic Coupon
数据结构·c++·算法·pat考试
AI科技星21 分钟前
质量定义方程的物理数学融合与求导验证
数据结构·人工智能·算法·机器学习·重构
精神病不行计算机不上班21 分钟前
[Java Web]在IDEA中完整实现Servlet的示例
java·servlet·tomcat·html·intellij-idea·web
chushiyunen23 分钟前
javadoc规范、idea生成javadoc等
java·ide