LeetCode279之完全平方数(相关话题：动态规划，四平方和定理)

数据与后端架构提升之路

于 2023-05-02 21:12:37 发布

阅读量709

点赞数

分类专栏： # 算法文章标签：动态规划算法数据结构

本文链接：https://blog.csdn.net/lzhcoder/article/details/130466659

版权

算法专栏收录该内容

125 篇文章 3 订阅

订阅专栏

题目描述

给你一个整数 n ，返回 和为 n 的完全平方数的最少数量 。

完全平方数 是一个整数，其值等于另一个整数的平方；换句话说，其值等于一个整数自乘的积。例如，1、4、9 和 16 都是完全平方数，而 3 和 11 不是。

示例 1：

输入：n = 12
输出：3 
解释：12 = 4 + 4 + 4

示例 2：

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

方法一（动态规划）

f[i]表示最少需要多少个数的平方来表示整数 i。

这些数必然落在区间 [1, $\sqrt{n}$ ]。我们可以枚举这些数，假设当前枚举到 j，那么我们还需要取若干数的平方，构成 $i-j^{2}$ 。此时我们发现

该子问题和原问题类似，只是规模变小了。这符合了动态规划的要求，于是我们可以写出状态转移方程。

$f|i|=1+\min_{j=1}^{\sqrt{i}}f|i-j^{2}|$

其中 f[0]=0 为边界条件，实际上我们无法表示数字 0，只是为了保证状态转移过程中遇到 j 恰为
$\sqrt{i}$ 的情况合法。同时因为计算 f[i]时所需要用到的状态仅有 $f|i-j^{2}|$ ，必然小于 i，因此我们只需要从小到大地枚举 i 来计算 f[i]即可。

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] = minn + 1;
        }
        return f[n];
    }
}

方法二（数学）

一个数学定理可以帮助解决本题：「四平方和定理」。

四平方和定理证明了任意一个正整数都可以被表示为至多四个正整数的平方和。这给出了本题的答案的上界。

同时四平方和定理包含了一个更强的结论：当且仅当 n≠4^k * (8m+7)时，n 可以被表示为至多三个正整数的平方和。因此，当 n=4^k * (8m+7)时nn 只能被表示为四个正整数的平方和。此时我们可以直接返回 4。

当 n≠4^k * (8m+7)时，我们需要判断到底多少个完全平方数能够表示 n，我们知道答案只会是 1,2,3中的一个：

答案为 1 时，则必有 nnn 为完全平方数，这很好判断；

答案为 2 时，则有 n=a^2+b^2 ，我们只需要枚举所有的 a( $1\leq a \leq \sqrt{n}$ )否为完全平方数即可；

答案为 3 时，我们很难在一个优秀的时间复杂度内解决它，但我们只需要检查答案为 1或 2 的两种情况，即可利用排除法确定答案。

class Solution {
    public int numSquares(int n) {
        if (isPerfectSquare(n)) {
            return 1;
        }
        if (checkAnswer4(n)) {
            return 4;
        }
        for (int i = 1; i * i <= n; i++) {
            int j = n - i * i;
            if (isPerfectSquare(j)) {
                return 2;
            }
        }
        return 3;
    }

    // 判断是否为完全平方数
    public boolean isPerfectSquare(int x) {
        int y = (int) Math.sqrt(x);
        return y * y == x;
    }

    // 判断是否能表示为 4^k*(8m+7)
    public boolean checkAnswer4(int x) {
        while (x % 4 == 0) {
            x /= 4;
        }
        return x % 8 == 7;
    }
}