LeetCode解题心得——完全平方数（python）

题目

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

思路

• 首先，我们准备小于给定数字 n 的完全平方数列表（即 square_nums）。
• 然后创建 queue 遍历，该变量将保存所有剩余项在每个级别的枚举。
• 在主循环中，我们迭代 queue 变量。在每次迭代中，我们检查余数是否是一个完全平方数。如果余数 不是一个完全平方数，就用其中一个完全平方数减去它，得到一个新余数，然后将新余数添加到 next_queue 中，以进行下一级的迭代。一旦遇到一个完全平方数的余数，我们就会跳出循环，这也意味着我们找到了解。

总的来说，将给定的正整数n看成是根节点，其与完全平方数列表中元素的差构成了它所有的子节点，子节点的子节点是其自身与完全平方数列表的差，以此类推。如果发现某个子节点在完全平方数列表中，则返回当前层数，正好使用层序遍历的宽度优先搜索

python代码

class Solution:
    def numSquares(self, n: int) -> int:
        squares_nums = [i**2 for i in range(1, int(n**0.5)+1)]
        queue = set()
        queue.add(n)
        nums = 0
        while queue:
            nums += 1
            next_queue = set()
            for res in queue:
                if res in squares_nums:
                        return nums
                for sq_num in squares_nums:
                    if res < sq_num:
                        break
                    else:
                        next_queue.add(res-sq_num)
            queue = next_queue
        return -1