python 用列表实现二叉树的宽度优先遍历

看了python中队列和栈的一些相关信息,看到很多人说列表其实就可以实现队列和栈的特点,手动试了一下,还真是,在此做一个记录,用列表来实现二叉树的按层级遍历

思路:将二叉树的根节点扔进列表中作为第一个元素,删除第一个元素并打印,同时将第一个元素的左节点和右节点添加到列表的尾部,循环如此,直至列表中所有的元素都打印完毕,代码实现如下:

def level(head):
    list = []
    if head:
        list.append(head)
    while len(list)>0:
        node = list.pop(0)
        print(node.data)
        if node.left:
            list.append(node.left)
        if node.right:
            list.append(node.right)

 

   

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章