力扣287——完全平方数(动态规划+贪心)

题目描述(中等)

给定正整数 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

思路

直接朴素的贪心会出错,就是由小于n的最大的完全平方数来组成n
最直接的例子就是12 方案应该是4+4+4而不是9+1+1+1,也就是不能贪心的去选9

所以想到动态规划,dp[i]是满足和为i的答案
对于i,可以由j 和 i-j,两部分组成
dp[i] = min(dp[i],所有可能的组合dp[j] + dp[i-j])
因为加上了对i的分割,这样复杂度O(n^2),TLE了

进行优化,在对i进行分隔的过程中,用一点贪心的思想
每次划分的过程中,步长不再是1,而是1 4 9 16…这样的完全平方数
步子迈大点,因为步子迈小了结果肯定只会大不会小,因为用了太多的1
这样复杂度就降到O(n sqrt(n))

另外官方题解的数学法真是长见识了,虽然在推导的过程中感觉到了答案应该就那几个数,没想到前人已经总结出来了。官方题解四平方和定理

代码

class Solution {
public:
    int numSquares(int n) {
        vector<int> dp(10005,6666666);
        for(int i = 1; i <= 100 ; i++){
            dp[i*i] = 1;
        }
        for(int i = 2; i <= n; i++){
            for(int j = i-1,k = 1;j>=i/2;j = i-k*k) {
                dp[i] = min(dp[i],dp[j] + dp[i-j]);
                k++;
            }
        }
        return dp[n];
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值