动态规划完全背包系列一＞完全平方数

用户11305962

发布于 2025-03-24 11:04:42

1170

文章被收录于专栏：学习学习

题目转化：

状态表示：

状态转移方程：

初始化：

填表顺序：

返回值：

代码呈现：

class Solution {

    public int numSquares(int n) {
        int aim = (int)Math.sqrt(n);
        int[][] dp = new int[aim+1][n+1];

        int INF = 0x3f3f3f3f;
        //初始化
        for(int i = 1; i <= n; i++) dp[0][i] = INF;
        
        for(int i = 1; i <= aim; i++)
            for(int j = 0; j <= n; j++){
                dp[i][j] = dp[i-1][j];
                if(j >= i*i)
                    dp[i][j] = Math.min(dp[i][j], dp[i][j-i*i]+1);
            }

        return dp[aim][n];    
    }
}

空间优化版本：

int aim = (int)Math.sqrt(n);
        int[] dp = new int[n+1];

        int INF = 0x3f3f3f3f;
        //初始化
        for(int i = 1; i <= n; i++) dp[i] = INF;
        
        for(int i = 1; i <= aim; i++)
            for(int j = i*i; j <= n; j++){
                dp[j] = Math.min(dp[j], dp[j-i*i] + 1);
            }

        return dp[n];

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2025-03-23，如有侵权请联系 cloudcommunity@tencent.com 删除

动态规划