社区所有版块导航
Python
python开源   Django   Python   DjangoApp   pycharm  
DATA
docker   Elasticsearch  
aigc
aigc   chatgpt  
WEB开发
linux   MongoDB   Redis   DATABASE   NGINX   其他Web框架   web工具   zookeeper   tornado   NoSql   Bootstrap   js   peewee   Git   bottle   IE   MQ   Jquery  
机器学习
机器学习算法  
Python88.com
反馈   公告   社区推广  
产品
短视频  
印度
印度  
Py学习  »  Python

每日一道算法题--leetcode 279--完全平方数--python

杉杉不要bug • 4 年前 • 415 次点击  
阅读 7

每日一道算法题--leetcode 279--完全平方数--python

【题目描述】

【方法一:动态规划】

自底向上填充dp,思路没问题,但是提交超时。

 def numSquares(self, n: int) -> int:
        dp=[i for i in range(n+1)]
        for i in range(4,n+1):
            for j in range(int(n**0.5),1,-1):
                if i>=j**2:
                    dp[i]=min(dp[i],dp[i-j*j]+1)
        return dp[n]
复制代码

【方法2:BFS】

相当于从根节点0,开始向下逐层延伸,每层可加的元素是1的平方到int(n**0.5)的平方,对于每个节点均如此。

from collections import deque
class Solution:
    def numSquares(self, n):
        selected=[i**2 for i in range(1,int(n**0.5)+1)]
        visited = set()
        queue=deque([(0,0)])
        while(queue):
            current,height=queue.popleft()
            for i in selected:
                sum_= current+i
                if sum_==n:
                    return height+1
                if sum_<n and sum_ not in visited:
                    visited.add(sum_)
                    queue.append((sum_,height+1))
复制代码
Python社区是高质量的Python/Django开发社区
本文地址:http://www.python88.com/topic/38431
 
415 次点击