【剑指offer】JZ4-重建二叉树

二叉树的问题一般都是分治思想，递归去做。因为二叉树本身就是递归定义的。

 

理论部分

何为前序遍历、中序遍历、后序遍历？

1.前序遍历：根左右

前序遍历的规则：根节点——左子树——右子树

2.中序遍历：左根右

中序遍历的规则：左子树——根节点——右子树

3.后序遍历：左右根

后序遍历的规则：左子树——右子树——根节点

 

前序遍历和中序遍历还原二叉树

思想：根据前序遍历找根——根据中序遍历找左右子树——递归前两步还原树

具体操作：

step1：根据前序遍历结果可知第一个元素为二叉树的根节点

step2：观察中序遍历结果，根节点的左侧为左子树，根节点的右侧为右子树，若根节点的左（右）侧无任何元素，则说明左（右）子树为空。

step3：递归上述过程。先找到当前树的根节点，然后划分为左右子树，进入左子树重复上述过程，进入右子树重复上述过程，最终还原二叉树。

 

例：

前序遍历：ABDHIEJKCFLMGNO

中序遍历：HDIBJEKALFMCNGO

按照上述步骤先画出二叉树，然后在进行求解后序遍历结果。结果为：HIDJKEBLMFNOGCA

 

前序遍历：GDAFEMHZ

中序遍历：ADEFGHMZ

求得后序遍历结果为：AEFDHZMG

 

前序遍历：ADCEFGHB

中序遍历：CDFEGHAB

求得后序遍历结果为：CFHGEDBA

 

中序遍历和后序遍历还原二叉树

思想：根据后序遍历找根——根据中序遍历找左右子树——递归前两步还原树

具体操作：

step1：根据后序遍历结果可知最后一个元素为二叉树的根节点

step2：观察中序遍历结果，根节点的左侧为左子树，根节点的右侧为右子树，若根节点的左（右）侧无任何元素，则说明左（右）子树为空

step3：递归上述过程。先找到当前树的根节点，然后划分为左右子树，进入左子树重复上述过程，进入右子树重复上述过程，最终还原二叉树。

中序遍历：HDIBJEKALFMCNGO

后序遍历：HIDJKEBLMFNOGCA

按照上述步骤先画出二叉树，然后在进行求解前序遍历结果。结果为：ABDHIEJKCFLMGNO

 

题解思路

 

方法一：递归法

代码

class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
        self.dic, self.po = {}, preorder
        for i in range(len(inorder)):
            self.dic[inorder[i]] = i
        return self.recur(0, 0, len(inorder) - 1)

    def recur(self, pre_root, in_left, in_right):
        if in_left > in_right: return # 终止条件：中序遍历为空
        root = TreeNode(self.po[pre_root]) # 建立当前子树的根节点
        i = self.dic[self.po[pre_root]]    # 搜索根节点在中序遍历中的索引，从而可对根节点、左子树、右子树完成划分。
        root.left = self.recur(pre_root + 1, in_left, i - 1) # 开启左子树的下层递归
        root.right = self.recur(i - in_left + pre_root + 1, i + 1, in_right) # 开启右子树的下层递归
        return root # 返回根节点，作为上层递归的左（右）子节点

 

详解注释版

#coding:utf-8

class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None
"""
问题：
用先序和中序重建二叉树

分析：
先序遍历的特点[根节点|左子树|右子树]
中序遍历的特点[左子树|根节点|右子树]

1.通过先序可以找出树的根节点
2.通过根节点和中序可以划分出树的左右子树

实现思路：
1.建立hash表，用来存储中序索引与元素的对应关系
2.之后用recur递归创建二叉树
"""

class Solution(object):
    def buildTree(self, in_order, pre_order):
        """

        :param in_order:
        :param pre_order:
        :return:
        """
        self.in_order_hash = {}
        self.pro= pre_order

        # 建立hash表
        in_order_length = len(in_order)
        for i in range(in_order_length):
            self.in_order_hash[in_order[i]] = i
        # 递归创建二叉树
        return self.recur(0, 0, in_order_length - 1)

    def recur(self, pre_root, in_left, in_right):
        """
        
        1. 边界条件：
        2.创建根节点:通过先序拿到
        3.去hash表中取根节点元素在中序中的对应索引
        4.构造出左右子树的先序根节点索引、中序左边界、中序右边界
        5.返回二叉树的根节点
        :param pre_root: 先序的根节点索引
        :param in_left: 中序的左边界
        :param in_right: 中序的右边界
        :return:
        """
        # 终止条件：中序遍历为空
        if in_left > in_right:
            return
        # 建立当前子树的根节点
        root = TreeNode(self.pro[pre_root])
        # 搜索根节点在中序遍历中的索引，从而可对根节点，左子树、右子树完成划分
        i = self.in_order_hash[root.val]
        # 左树---开启左子树的下层递归
        root.left = self.recur(pre_root + 1, in_left, i - 1)
        # 右树---开启右子树的下层递归
        # 先序中的根节点索引为：根节点+左子树+1
        root.right = self.recur(pre_root + i - in_left + 1, i + 1, in_right)
        # 返回根节点，作为上层递归的左（右）子节点
        return root

 

 

Reference

1.二叉树的前序遍历、中序遍历和后序遍历之间还原二叉树

2.递归法题解思路