279. 完全平方数——最短路径问题

算法中用到最少这个关键字——最短路径问题。

最短路径算法其实就是图的广度优先遍历。

思路参考:Link

class Solution(object):
    def numSquares(self, n):
        """
        :type n: int
        :rtype: int
        """
        # python开根号 num_sqrt = n**0.5
        # 对图中所有的点进行预处理,全都没被访问过,设置为false.
        q = []
        q.append((n,0))
        visited = [False for _ in range(n+1)]
        visited[n] = True
        print(q)

        while any(q): # 如果队列不为空的话,执行循环
            num, step = q.pop(0)  # 队列先进先出

            i = 1
            tNum = num-i**2
            while tNum>=0:
                if tNum==0:
                    return step+1

                if not visited[tNum]: # 如果图中的节点没有被访问过
                    q.append((tNum,step+1))
                    visited[tNum]=True
                    print(q)

                i+=1
                tNum = num-i**2

猜你喜欢

转载自blog.csdn.net/weixin_31866177/article/details/83901463