【力扣279. 完全平方数】完全背包+数学法-四平方和定理+BFS（python3）

最新推荐文章于 2022-09-04 18:21:29 发布

Alwaysion

最新推荐文章于 2022-09-04 18:21:29 发布

阅读量187

点赞数

分类专栏： # BFS-树 # 动态规划 # 基本算法文章标签： leetcode 动态规划 python 背包 bfs

本文链接：https://blog.csdn.net/anan15151529/article/details/118308465

版权

动态规划同时被 3 个专栏收录

31 篇文章 1 订阅

订阅专栏

基本算法

29 篇文章 1 订阅

订阅专栏

BFS-树

7 篇文章 0 订阅

订阅专栏

题目描述

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

思路题解

完全背包

一开始的思路：n=6665的时候，时间超限

class Solution:
    def numSquares(self, n: int) -> int:
        if n<4:return n
        dp=[10001]*(n+1)
        dp[1],dp[2],dp[3],dp[4]=1,2,3,1
        for i in range(5,n+1):
            if math.sqrt(i)%1!=0:
                for j in range(1,i//2+1):
                    dp[i]=min(dp[i-j]+dp[i-(i-j)],dp[i])
            else:dp[i]=1
        return dp[n]

看了下答案，进行了改进

class Solution:
    def numSquares(self, n: int) -> int:
        if n<4:return n
        dp=[i for i in range(n+1)]
        for i in range(2,n+1):
            j=1
            while j*j<=i:
                dp[i]=min(dp[i-j*j],dp[i])
                j+=1
            dp[i]+=1
        return dp[n]