LeetCode-Python-279. 完全平方数

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

示例 1:

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

示例 2:

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

思路：

依然是一个求最短路径的问题，每一条路径的起点是当前的和，下一个节点是加上一个完全平方数之后的和，求从0到n最短的移动步数，所以用BFS处理。

from collections import deque
class Solution(object):
    def numSquares(self, n):
        """
        :type n: int
        :rtype: int
        """
        #起始点：当前的和
        #下一层：加上一个完全平方数之后的和
        #求最短路径用BFS
        record = []
        for i in range(1, int(n ** 0.5) + 1):
            record.append(i * i)
        # print record
        visited = set()
        q = deque()
        q.append([0, 0])
        while(q):
            m, cnt = q.popleft()
            
            for num in record:
                s = m + num
                if s == n:
                    return cnt + 1
                if s < n and s not in visited:
                    visited.add(s)
                    q.append([s, cnt + 1])