leetcode hot100 之 完全平方数

题目

给定一个数 n，返回和为 n 的完全平方数的最少数量。
完全平方数指的是 一个整数，其值正好等于另一个整数的平方。如 1、4、9、16、25 等。

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

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

思路

思路一

动态规划。先上公式再解释
$$dp[i]=1+\underset{j=1}{\overset{\sqrt{i}}{\min }}dp[i-j\ast j]$$
对于数字n，如果有某个数字 j，j * j 小于 n，那么 n 可以拆分成 j * j + ( n - j * j)，如果我们知道剩余部分 ( n - j * j) 的完全平方数最少数量，n 拆成 j * j + ( n - j * j) 的完全平方数最少数量。
而 j 的取值范围为 (1, sqrt(n))。而 数字 n 的完全平方数的最少数量则对应某一个 j 的组合。

• 复杂度分析
  • 时间复杂度 O(n*sqrt(n))。
  • 空间复杂度 O(n)。

思路2

还有一种时间和空间复杂度都更低的方法，需要一些数学上的推导。这里就不展开了，感兴趣的可以了解：四平方和定理。

代码

class Solution {
public:
    int numSquares(int n) {
        // dp[i] = 1 + min^{sqrt_i}_{j} dp(i - j*j)
        vector<int> dp(n+1);
        dp[0] = 0;
        for (int i = 1; i <= n; i++) {
            int min_num = INT_MAX;
            for (int j = 1; j * j <= i; j++) {
                min_num = min(min_num, dp[i - j * j]);
            }
            dp[i] = 1 + min_num;
        }
        return dp[n];
    }
};