二叉树四种遍历方法

ぃ、小莉子 提交于 2020-02-26 11:01:27
class Node(object):
    def __init__(self, item):
        self.item = item
        self.lchild = None
        self.rchild = None


class Tree(object):
    """二叉树"""

    def __init__(self):
        self.root = None

    def add(self, item):
        node = Node(item)
        if self.root is None:
            self.root = node
            return None
        queue = [self.root]
        while queue:
            cur = queue.pop(0)
            if cur.lchild is None:
                cur.lchild = node
                return None
            else:
                queue.append(cur.lchild)
            if cur.rchild is None:
                cur.rchild = node
                return None
            else:
                queue.append(cur.rchild)

    def breadth_travel(self):
        """广度遍历"""
        queue = [self.root]
        if self.root is None:
            return None
        while queue:
            cur_node = queue.pop(0)
            print(cur_node.item)
            if cur_node.lchild is not None:
                queue.append(cur_node.lchild)
            if cur_node.rchild is not None:
                queue.append(cur_node.rchild)

    def preorder(self, node):
        """前序遍历"""
        if node is None:
            return None
        print(node.item)
        self.preorder(node.lchild)
        self.preorder(node.rchild)

    def inorder(self, node):
        """中序遍历"""
        if node is None:
            return None
        self.inorder(node.lchild)
        print(node.item)
        self.inorder(node.rchild)

    def postorder(self, node):
        """后序遍历"""
        if node is None:
            return None
        self.postorder(node.lchild)
        self.postorder(node.rchild)
        print(node.item)

广度遍历通过队列这种数据结构来实现。
三种深度遍历中都使用了递归的思想。其中前序遍历比较好理解,中序和后续自己一开始没想明白,以为是从左下的叶子节点开始遍历,当然遍历的结果确实是左下叶子节点排在最前面。但是正确的想法应该是从根节点开始,一步一步分割左子树和右子树,然后按照顺序要求写出最后的结果。自己一开始想复杂了。其实用递归非常容易实现。
还有一个小知识点,在三种深度遍历模式中,如果想通过遍历的结果得到树的具体结构,则至少需要两种遍历的结果,并且其中一种必须为中序遍历。通过前序遍历或者后序遍历结合中序遍历得到包括根节点在内的各个节点,再重构整棵树的结构

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!