二叉树的重构（Python实现）

二叉树的重构

1. 概念

   二叉树的重构 是指通过二叉树的遍历结果得到二叉树的构造。

    重构方案：    前序遍历(  PreOrder) + 中序遍历(InOrder)
                            后序遍历(PostOrder) + 中序遍历(InOrder)
                            前序遍历(  PreOrder) + 后序遍历(PostOrder)（只对真二叉树成立，其他的情况不能重构出一棵二叉树）

 

2. 重构二叉树实现

题目要求：输入前序遍历和中序遍历，重构二叉树，并输出后序遍历。

算法设计思想：
1.找出前序遍历中第一个值，即根节点，并记录它在中序遍历中的位置。
2.根据根节点，将中序遍历分为左子树中序遍历和右子树中序遍历，并根据1中记录的位置分别得到子树的前序遍历。
3.判断子树的前、中序遍历是否为空，是返回None，否递归调用1,2步。
4.返回二叉树。

5.遍历输出后序序列

代码实现：

#1.前序和中序序列重建一颗二叉树
class TreeNode(object):
    def __init__(self, elem):
        self.elem = elem
        self.lchild = None
        self.rchild = None


class TreeHandler():
    # 重构二叉树
    def reConsTree(self, pre, mid):
        # 判断前、中序遍历是否为空
        if (len(pre) or len(mid)) == 0:
            return
        midindex = mid.index(pre[0]) # 根节点在mid中的位置
        tree = TreeNode(pre[0]) # 实例化二叉树对象
        # 根据前、中序遍历对左、右子树递归构造
        tree.left=self.reConsTree(pre[1:midindex+1],mid[:midindex])
        tree.right=self.reConsTree(pre[midindex+1:],mid[midindex+1:])
        return tree
    # 后序遍历二叉树
    def traversalTreeNode(self,tree):
        if tree != None:
            self.traversalTreeNode(tree.left)
            self.traversalTreeNode(tree.right)
            print(tree.elem,end=' ')



pre = [4, 2, 1, 3, 5, 7, 6]
mid = [1, 2, 3, 4, 5, 6, 7]
Btree = TreeHandler()
tree = Btree.reConsTree(pre,mid)
Btree.traversalTreeNode(tree)

输出结果：

1 3 2 6 7 5 4