[算法题] 重建二叉树

Posted 2020-10-02

题目描述

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

题目思路

本题就是按照建立二叉树的思路建立就行了。先序遍历的第一个是根节点，然后在中序遍历找到该根节点，以此为界，中序遍历的左边是它的左子树的中序遍历，同样地找到该左子树在先序遍历中对应的先序遍历顺序。对于右子树也是一样的方法。

本体采用递归，递归就要先写出终止条件。

Python代码

这个题目用Python非常方便，因为可以直接利用到Python中的切片技术，省时省力通俗易懂。Java版本相对复杂得多。

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回构造的TreeNode根节点
    def reConstructBinaryTree(self, pre, tin):
        # write code here
        if len(pre)==0 or len(tin)==0:
            return
        if len(pre)==1:
            return TreeNode(pre[0])
        node=TreeNode(pre[0])
        for i in range(len(tin)):
            if tin[i]==pre[0]:
                break
        node.left=self.reConstructBinaryTree(pre[1:i+1],tin[:i])
        node.right=self.reConstructBinaryTree(pre[i+1:],tin[i+1:])
        return node