【leetcode】279 完全平方数（动态规划）

题目链接：https://leetcode-cn.com/problems/perfect-squares/

题目描述

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

示例 1:

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

示例 2:

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

思路

1 动态规划

时间复杂度：$O(n^{3/2})$
空间复杂度：$O(n)$

维护一个长度为n+1的数组dp，dp[i]表示i的最小完全平方数组成个数；
for i in 1~n：
（1）如果i为完全平方数，有dp[i] = 1
（2）如果i不为完全平方数，dp[i] = min(dp[j*j] + dp[i - j*j]) for j = 1~sqrt(n)

/*
 * 完全平方数-动态规划
 * 时间复杂度O(N^(3/2)) 空间复杂度O(N)
 */
class Solution {
public:
    int numSquares(int n) {
        if(n <= 0) return 0;
        if(isSquare(n)) return 1;

        vector<int> dp(n+1, 0);

        for (int i = 1; i <=n ; ++i) {  // 注意对完全平方数的定义从1开始
            if(isSquare(i)){
                dp[i] = 1;  // 完全平方数
                continue;
            }
            int tmp = int(sqrt(i));
            int minCount = n;
            // 在比i小的数中寻找最少组合
            for (int j = 1 ; j*j < i ; ++j) {
                minCount = min(minCount,dp[j*j] + dp[i - j*j]);
            }
            dp[i] = minCount;
        }
        return dp[n];
    }

private:
    // 判断是否为完全平方数
    bool isSquare(int n){
        int tmp = int(sqrt(n));
        return n == tmp * tmp;
    }
};

2 数学公式

拉格朗日四平方数和定理：
任何一个正整数都可以表示成不超过四个整数的平方之和。 推论：满足四数平方和定理的数n（四个整数的情况），必定满足 $n=4^a(8b+7)$

/*
 * 拉普拉斯四平方数和公式
 */
class SolutionI {
public:
    int numSquares(int n) {
        //先根据上面提到的公式来缩小n
        while(n % 4 == 0) {
            n /= 4;
        }
        //如果满足公式 则返回4
        if(n % 8 == 7) {
            return 4;
        }
        //在判断缩小后的数是否可以由一个数的平方或者两个数平方的和组成
        int a = 0;
        while ((a * a) <= n) {
            int b = sqrt((n - a * a));
            if(a * a + b * b == n) {
                //如果可以 在这里返回
                if(a != 0 && b != 0) {
                    return 2;
                } else{
                    return 1;
                }
            }
            a++;
        }
        //如果不行 返回3
        return 3;
    }
};