LeetCode #279 完全平方数 动态规划 BFS

LeetCode #279 完全平方数

题目描述

给定正整数 n，找到若干个完全平方数（比如 ·1, 4, 9, 16, …·）使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。

示例 1:

输入: n = 12
输出: 3 
解释: 12 = 4 + 4 + 4.

示例 2:

输入: n = 13
输出: 2
解释: 13 = 4 + 9.

方法一：动态规划

• 初始化长度为 n+1 的数组 dp，每个位置都为（对应下标也行，后面就不用赋值）
• 如果 n 为 0，则结果为 0
• 对数组进行遍历，下标为 i，每次先将当前位置更新为最坏结果，即 dp[i]=i，如 4=1+1+1+1
• 动态转移方程为：dp[i] = min(dp[i], dp[i-j*j] + 1)

class Solution:
    def numSquares(self, n: int) -> int:
    	# 这里直接对dp赋值
        dp=[i for i in range(n+1)]
        for i in range(2,n+1):
            for j in range(1,int(i**(0.5))+1):
                dp[i]=min(dp[i],dp[i-j*j]+1)
        return dp[-1]

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

方法二：BFS

class Solution:
    def numSquares(self, n: int) -> int:
        if n == 0: return n
        queue = [(n, 0)]
        # 最开始用列表超时了
        # visited = []
        visited = set()
        while queue:
            x, step = queue.pop(0)
            for i in range(1, int(x**0.5) + 1):
                j = x - i * i
                if j == 0:
                    return step + 1
                if not j in visited:
                    queue.append((j, step + 1))
                    visited.add(j)
        return n