如何利用python实现二叉树？附代码讲解_python定义二叉树的结构(2)

def level_travel(self):    if self.root == None:        return     queue = [self.root]    while queue:        cur = node = queue.pop(0)        print(cur.elem, end=" ")        if cur.lchild != None:            queue.append(cur.lchild)        if cur.rchild != None:            queue.append(cur.rchild)

添加节点

树具有一个层次结构，我们在每层按照从左至右的方向添加节点。我们从第一层开始，依次判断根节点是否有左子节点和右子节点，如果没有，则添加；如果有，则转到第二层。以此类推。实现上与层遍历有点类似。

def add(self, item):    node = Node(item)    # 如果节点为空    if self.root == None:        self.root = node        return    queue = [self.root]    while queue:        cur = queue.pop(0)        if cur.lchild == None:            cur.lchild = node            return         else:            queue.append(cur.lchild)        if cur.rchild == None:            cur.rchild = node            return         else:            queue.append(cur.rchild)

备注：添加节点while queue:

cur = node = queue.pop(0)，不能把queue.pop(0)的值赋给node, 因为node = Node(item), 如果赋给了node, 就会让node无限指向自己，应该写成cur = queue.pop(0)

先序遍历

先序遍历是先遍历根节点，再遍历左子树，最后遍历右子树。对于遍历子树而言，也是遍历子树的根节点，再遍历子树的左子树，最后遍历子树的右子树。可以用递归实现。

def preorder_travel(self, node):
    # 如果节点为空
    if node == None:
        return

    print(node.elem， end=" ")
    self.preorder_travel(node.lchild)
    self.preorder_travel(node.rchild)

中序遍历

中序遍历则先遍历左子树，再遍历根节点，最后遍历右子树。

def inorder_travel(self, node):
    # 如果节点为空
    if node == None:
        return
    self.inorder_travel(node.lchild)
    print(node.elem, end=" ")
    self.inorder_travel(node.rchild)

后序遍历

后续遍历先遍历左子树，再遍历右子树，最后遍历根节点。

def postorder_travel(self, node):
    # 如果节点为空
    if node == None:
        return

    self.postorder_travel(node.lchild)
    self.postorder_travel(node.rchild)
    print(node.elem, end=" ")

测试

if __name__ == "__main__":    t = Tree()    t.add(0)    t.add(1)    t.add(2)    t.add(3)    t.add(4)    t.add(5)    t.add(6)    t.add(7)    t.add(8)    t.add(9)    t.level_travel()    print()    t.preorder_travel(t.root)    print()    t.inorder_travel(t.root)    print()    t.postorder_travel(t.root)    print()

今天的小知识学会了么？

---

现在能在网上找到很多很多的学习资源，有免费的也有收费的，当我拿到1套比较全的学习资源之前，我并没着急去看第1节，我而是去审视这套资源是否值得学习，有时候也会去问一些学长的意见，如果可以之后，我会对这套学习资源做1个学习计划，我的学习计划主要包括规划图和学习进度表。

分享给大家这份我薅到的免费视频资料，质量还不错，大家可以跟着学习

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化学习资料的朋友，可以戳这里无偿获取

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！