[leetcode Perfect Squares] 最少完全平方数

原题:

给定一个正整数n,要求找出最小数量的完全平方数,使得它们的和等于n。

既然是最小的数量,那就是完全平方数需要尽可能大,我的第一反应是能不能使用贪心算法。比如给定数字13,第一个最大的完全平方数是9,还剩下4刚好也是一个完全平方数。那么,贪心算法真的在任何情况下都能使用吗?我们不妨再看一个例子,比如说12:根据贪心原则,第一个最大的完全平方数是9,此时还剩下3,3只能对应3个1。也就是说我们总共用了4个数,但是我们发现12 = 4 + 4 + 4,只需要使用3个数。因此,贪心算法在这里并不适用。

我们不妨先看下如何用暴力的方法来解决它。还是以刚才的数字12为例:第一个数先取9,12减9还剩3,然后继续对3求最小数量的完全平方数。也就是说,我们将原本一个大问题转化为了一个更小的问题,运用了“分而治之”的思想,很容易想到可以使用递归。若一个数取9无法完成,接下来再用4去尝试。找出所有的可行解,并在可行解中返回数量最少的那个。

但是,如果按照这种方式,算法的复杂度将会变得非常大。不知道大家有没有想到在用递归求斐波那契数列的时候,会产生很多冗余的运算。我们可以使用一个数组来保存中间计算的结果。每次递归的时候,若能在数组中找到解,则直接返回,不需要再次运算。在这道题中可以使用类似的方法,我们用dp[i - 1]表示数字i最少需要多少个完全平方数,通过保留状态的方法,这其实就是动态规划。

另外,我们还可以对算法做一个剪枝操作,我们不需要每次从最大的完全平方数一直遍历到1,只要遍历前半部分就可以(为什么?自己想)。比如说数字17,最大的完全平方数是16,我们只需要从4遍历到3即可。

最后,根据上面的分析,我们的代码如下:

class Solution {
public:
    int numSquares(int n) {
        int dp[n];
        memset(dp, 0, n * sizeof(n));
        int res = dfs(n, 0, dp);
        return res;
    }

private:
    /*
     * 用count记录当前使用了多少个完全平方数
     */
    int dfs(int n, int count, int *dp) {

        if (n == 0) { //已经数字n分解完,直接返回count
            return count;
        }

        int c = 0;
        if ((c = dp[n - 1]) != 0) { //若在状态数组中能找到解,直接返回,
            return count + c;       //并且还需要加上已经使用的个数
        }
        
        //从所有的解中找出使用数字最少的
        int res = INT_MAX;
        int j = (int) sqrt(n);
        for (int i = j; i >= (j / 2 + 1); i--) {
            int num = n - pow(i, 2); 
            int c = dfs(num, count + 1, dp);
            res = min(res, c);
        }
        dp[n - 1] = res - count; //计算完成后,将结果保存到状态数组中
        return res;
    }

};

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • LeetCode 刷题随手记 - 第一部分 前 256 题(非会员),仅算法题,的吐槽 https://leetc...
    蕾娜漢默阅读 18,003评论 2 36
  • 此笔记用于指导初学者阅读。原文在此!,出于便于交流的考虑,内容重放在此。由于作业部落支持LaTex公式,所以,更加...
    Bintou老师阅读 12,333评论 0 27
  • 操作系统 线程和进程的区别(1)地址空间:进程内的一个执行单元;进程至少有一个线程;它们共享进程的地址空间;而进程...
    Stephen__Li阅读 642评论 0 1
  • ▼ 说起日料眼前总会浮现出电影《奔爱》中的场面 章子怡饰演的电影女主与男主彭于晏相识于日本小樽 男主用了三年时间才...
    口水秀阅读 338评论 0 0
  • 多想写封信给你 向你倾诉这些年的苦乐与悲喜 让你听到我的哀愁与叹息 期待着你会说一句:照顾好自己 多想写封信给你 ...
    鲸霞阅读 154评论 5 9