python数据结构与算法----二叉树

二叉树的基本概念

二叉树是每个节点最多有两个子树的树结构。通常子树被称作“左子树”(left subtree)和“右子树”(right subtree)

二叉树的性质(特性)

性质1: 在二叉树的第i层上至多有2^(i-1)个结点(i>0)
性质2: 深度为k的二叉树至多有2^k - 1个结点(k>0)
性质3: 对于任意一棵二叉树,如果其叶结点数为N0,而度数为2的结点总数为N2,则N0=N2+1;
性质4:具有n个结点的完全二叉树的深度必为 log2(n+1)
性质5:对完全二叉树,若从上至下、从左至右编号,则编号为i 的结点,其左孩子编号必为2i,其右孩子编号必为2i+1;其双亲的编号必为i/2(i=1 时为根,除外)

(1)完全二叉树——若设二叉树的高度为h,除第 h 层外,其它各层 (1~h-1) 的结点数都达到最大个数,第h层有叶子结点,并且叶子结点都是从左到右依次排布,这就是完全二叉树。


(2)满二叉树——除了叶结点外每一个结点都有左右子叶且叶子结点都处在最底层的二叉树。

二叉树的节点表示以及树的创建

通过使用Node类中定义三个属性,分别为elem本身的值,还有lchild左孩子和rchild右孩子

#构造节点类
class Node():
    def __init__(self,item):
        self.elem=item
        self.lchild=None
        self.rchild=None

树的创建,创建一个树的类,并给一个root根节点,一开始为空,随后添加节点

#构造树类
class Tree():
    def __init__(self):
        self.root=None
    #添加元素
    def add(self,item):
        node=Node(item)
        queue = [self.root]
        # 如果树是空的,则对根节点赋值
        if self.root is None:
            self.root=node
            return
        # 对已有的节点进行层次遍历
        while queue:
            # 弹出队列的第一个元素
            cur_queue=queue.pop(0)
            # 如果左右子树都不为空,加入队列继续判断
            if cur_queue.lchild is None:
                cur_queue.lchild=node
                return
            else:
                queue.append(cur_queue.lchild)
            if cur_queue.rchild is None:
                cur_queue.rchild=node
                return
            else:
                queue.append(cur_queue.rchild)

二叉树的遍历

树的遍历是树的一种重要的运算。所谓遍历是指对树中所有结点的信息的访问,即依次对树中每个结点访问一次且仅访问一次,我们把这种对所有节点的访问称为遍历(traversal)。那么树的两种重要的遍历模式是深度优先遍历和广度优先遍历,深度优先一般用递归,广度优先一般用队列。一般情况下能用递归实现的算法大部分也能用堆栈来实现。

深度优先遍历

对于一颗二叉树,深度优先搜索(Depth First Search)是沿着树的深度遍历树的节点,尽可能深的搜索树的分支。
那么深度遍历有重要的三种方法。这三种方式常被用于访问树的节点,它们之间的不同在于访问每个节点的次序不同。这三种遍历分别叫做先序遍历(preorder),中序遍历(inorder)和后序遍历(postorder)。我们来给出它们的详细定义,然后举例看看它们的应用。

  • 先序遍历 在先序遍历中,我们先访问根节点,然后递归使用先序遍历访问左子树,再递归使用先序遍历访问右子树
    根节点->左子树->右子树
  • 中序遍历 在中序遍历中,我们递归使用中序遍历访问左子树,然后访问根节点,最后再递归使用中序遍历访问右子树
    左子树->根节点->右子树
  • 后序遍历 在后序遍历中,我们先递归使用后序遍历访问左子树和右子树,最后访问根节点
    左子树->右子树->根节点
#深度优先遍历
    '''
    先序遍历:根左右
    中序遍历:左根右
    后序遍历:左右根
    '''
    #递归实现先序遍历:根 -->左 -->右
    def preorder(self,node):
        if node is None:
            return
        print(node.elem,end=' ')
        self.preorder(node.lchild)
        self.preorder(node.rchild)

    # 递归实现中序遍历:左 -->根 -->右
    def inorder(self,node):
        if node is None:
            return
        self.inorder(node.lchild)
        print(node.elem,end=' ')
        self.inorder(node.rchild)

    # 递归实现后序遍历:左 -->右 -->根
    def postorder(self,node):
        if node is None:
            return
        self.postorder(node.lchild)
        self.postorder(node.rchild)
        print(node.elem,end=' ')


广度优先遍历(层次遍历)

从树的root开始,从上到下从从左到右遍历整个树的节点

#广度优先遍历
    def breadth_travel(self):
        """利用队列实现树的层次遍历"""
        if self.root is None:
            return
        queue=[self.root]
        while queue:
            node=queue.pop(0)
            print(node.elem,end=' ')
            if node.lchild is not None:
                queue.append(node.lchild)
            if node.rchild is not None:
                queue.append(node.rchild)

二叉树的遍历整体代码:

#构造节点类
class Node():
    def __init__(self,item):
        self.elem=item
        self.lchild=None
        self.rchild=None
#构造树类
class Tree():
    def __init__(self):
        self.root=None
    #添加元素
    def add(self,item):
        node=Node(item)
        queue = [self.root]
        # 如果树是空的,则对根节点赋值
        if self.root is None:
            self.root=node
            return
        # 对已有的节点进行层次遍历
        while queue:
            # 弹出队列的第一个元素
            cur_queue=queue.pop(0)
            # 如果左右子树都不为空,加入队列继续判断
            if cur_queue.lchild is None:
                cur_queue.lchild=node
                return
            else:
                queue.append(cur_queue.lchild)
            if cur_queue.rchild is None:
                cur_queue.rchild=node
                return
            else:
                queue.append(cur_queue.rchild)

    #广度优先遍历
    def breadth_travel(self):
        """利用队列实现树的层次遍历"""
        if self.root is None:
            return
        queue=[self.root]
        while queue:
            node=queue.pop(0)
            print(node.elem,end=' ')
            if node.lchild is not None:
                queue.append(node.lchild)
            if node.rchild is not None:
                queue.append(node.rchild)

    #深度优先遍历
    '''
    先序遍历:根左右
    中序遍历:左根右
    后序遍历:左右根
    '''
    #递归实现先序遍历:根 -->左 -->右
    def preorder(self,node):
        if node is None:
            return
        print(node.elem,end=' ')
        self.preorder(node.lchild)
        self.preorder(node.rchild)

    # 递归实现中序遍历:左 -->根 -->右
    def inorder(self,node):
        if node is None:
            return
        self.inorder(node.lchild)
        print(node.elem,end=' ')
        self.inorder(node.rchild)

    # 递归实现后序遍历:左 -->右 -->根
    def postorder(self,node):
        if node is None:
            return
        self.postorder(node.lchild)
        self.postorder(node.rchild)
        print(node.elem,end=' ')

if __name__ == '__main__':
    tree=Tree()
    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.breadth_travel()
    print()
    print('------先序遍历------')
    tree.preorder(tree.root)
    print()
    print('------中序遍历------')
    tree.inorder(tree.root)
    print()
    print('------后序遍历------')
    tree.postorder(tree.root)
----广度优先遍历----
0 1 2 3 4 5 6 7 8 9 
------先序遍历------
0 1 3 7 8 4 9 2 5 6 
------中序遍历------
7 3 8 1 9 4 0 5 2 6 
------后序遍历------
7 8 3 9 4 1 5 6 2 0 




猜你喜欢

转载自blog.csdn.net/weixin_39881922/article/details/79750206