【算法】二叉树的广度遍历

#encoding=utf-8

from queue import Queue

class TreeNode(object):
    def __init__(self,val,left=None,right=None):
        self.val=val
        self.left=left
        self.right=right

class BinaryTree(object):
    def __init__(self,root=None):
        self.root=root #这里传入的是创建好的Tree
        #print(root.val)

    def breathSearth(self):
        if self.root==None:
            return None
        retlist=[]
        queue=Queue()
        queue.put(self.root) #用队列的方式，先进先出，先将Tree从顶到底一次放进去，然后调用queue.get()方法，依次取出来存到retlist中
        while queue.empty() is not True:
            node=queue.get()
            retlist.append(node.val)
            if node.left!=None: #如果左节点和右节点不为空，再依次放到队列中，while循环再取出来存到retlist中
                queue.put(node.left) 
            if node.right!=None:
                queue.put(node.right)
        return retlist

if __name__=="__main__":
    #创建二叉树
    rootNode=TreeNode(50)
    rootNode.left=TreeNode(20,left=TreeNode(15),right=TreeNode(30))
    rootNode.right=TreeNode(60,right=TreeNode(70))

    #实例化
    tree=BinaryTree(rootNode) #把创建好的二叉树传进去
    retlist=tree.breathSearth()
    print(retlist)