每日一练 | Day 10

279. 完全平方数

题目链接

https://leetcode.cn/problems/perfect-squares/

相关算法

动态规划

题目描述

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

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

• 数据范围：
  1 <= n <= 10^4

解题思路

本题可以使用背包问题的思路来进行求解，即有若干大小为一个整数的平方的物品，如1、4、9 和 16 等，要使用这些物品填满大小为 n 的空间，最少放置多少个物品。我选择使用动态规划进行求解，先定义 一个长度为 n + 1 的 dp 数组，dp[i] 要填满大小为 i 的空间，至少要使用多少物品。表示因为要求的是最小值，所以先使用 Arrays.fill(dp, Integer.MAX_VALUE) 将数组所有元素初始化为最大值，同时设置初始值 dp[0] = 0。然后就可以开始考虑递推公式了，我们可以枚举从 1 到 $\sqrt{i}$ 的数字，使用 j 来表示，则有 dp[i] = Math.min(dp[i], dp[i - j * j] + 1)，至此逻辑梳理完成

完整代码

class Solution {
    public int numSquares(int n) {
        int[] dp = new int[n + 1];
        Arrays.fill(dp, Integer.MAX_VALUE);
        dp[0] = 0;
        for (int i = 1; i < dp.length; i++) {
            for (int j = 0; j * j <= n; j++) {
                if (i - j * j >= 0 && dp[i - j * j] != Integer.MAX_VALUE) {
                    dp[i] = Math.min(dp[i], dp[i - j * j] + 1);
                }
            }
        }
        return dp[n];
    }
}

• 时间复杂度：$O(n\sqrt{n})$
• 空间复杂度：$O(n)$