笔记本📒

python-完全二叉树的广度优先遍历-先序遍历-中序遍历-后序

2021-02-04  本文已影响0人  涓涓自然卷
完全二叉树.png

一、广度优先遍历:对每一层节点依次访问,访问完一层进入下一层,而且每个节点只能访问一次。对于例子来说,广度优先遍历结果为:0123456789
二、先序遍历:依次先序遍历根、左、右。对于例子来说,先序遍历结果为:0137849256
三、中序遍历:依次中序遍历左、根、右。对于例子来说,中序遍历结果为:7381940526
四、后序遍历:依次后序遍历左、右、根。对于例子来说,后序遍历结果为:7839415620

代码:
class Node(object):
    """节点类"""

    def __init__(self, item):
        self.item = item
        self.lchild = None
        self.rchild = None


class BinaryTree(object):
    """完全二叉树"""

    def __init__(self, node=None):
        self.root = node

    def add(self, item):
        """添加节点"""

        if self.root == None:
            self.root = Node(item)
            return

        # 队列
        queue = []
        # 从尾部添加数据
        queue.append(self.root)

        while True:
            # 从头部取出数据
            node = queue.pop(0)
            # 判断左节点是否为空
            if node.lchild == None:
                node.lchild = Node(item)
                return
            else:
                queue.append(node.lchild)

            if node.rchild == None:
                node.rchild = Node(item)
                return
            else:
                queue.append(node.rchild)

    def breadh_travel(self):
        """广度优先遍历"""
        # 判断根节点是否为空
        if self.root == None:
            return

        # 队列
        queue = []
        # 添加数据
        queue.append(self.root)

        while len(queue) > 0:
            # 取出数据
            node = queue.pop(0)
            print(node.item, end="")

            # 判断左右子节点是否为空
            if node.lchild is not None:
                queue.append(node.lchild)
            if node.rchild is not None:
                queue.append(node.rchild)

    def preorder_travel(self, root):
        """先序遍历 根 左 右"""
        if root is not None:
            print(root.item, end="")
            self.preorder_travel(root.lchild)
            self.preorder_travel(root.rchild)

    def inorder_travel(self, root):
        """中序遍历 左 根 右"""
        if root is not None:
            self.inorder_travel(root.lchild)
            print(root.item, end="")
            self.inorder_travel(root.rchild)

    def postorder_travel(self, root):
        """后序遍历 根 左 右"""
        if root is not None:
            self.postorder_travel(root.lchild)
            self.postorder_travel(root.rchild)
            print(root.item, end="")


if __name__ == '__main__':
    tree = BinaryTree()
    tree.add("0")
    tree.add("1")
    tree.add("2")
    tree.add("3")
    tree.add("4")
    tree.add("5")
    tree.add("6")
    tree.add("7")
    tree.add("8")
    tree.add("9")
    print("广度优先遍历结果:")
    tree.breadh_travel()
    print()
    print("先序遍历结果:")
    tree.preorder_travel(tree.root)
    print()
    print("中序遍历结果:")
    tree.inorder_travel(tree.root)
    print()
    print("后续遍历结果:")
    tree.postorder_travel(tree.root)
    print()


执行结果:
运行结果.png
上一篇 下一篇

猜你喜欢

热点阅读