输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
树是一种常见数据结构。除根结点外每个结点只有一个父结点,根结点没有父结点,除叶结点外所有结点都有一个或多个子结点,叶结点没有子结点。父结点与子结点之间用指针链接。树的操作涉及大量指针。
面试时提到的树大多是二叉树。二叉树每个结点最多只能有两个子结点。
二叉树中最重要的操作:遍历,即按某一顺序访问树中的所有结点。遍历方式如下三种:
- 前序遍历:先访问根结点,再访问左子结点,最后访问右子结点
- 中序遍历:先访问左子结点,再访问根结点,最后访问右子结点
- 后序遍历:先访问左子结点,再访问右子结点,最后访问根结点
这三种遍历方式都递归和循环两种不同的实现方法。每种遍历的递归实现都比循环实现简捷很多
分析:
二叉树的前序遍历中,第一个数字总是树的根结点的值。中序遍历中,根结点值在序列中间,根结点左边均为左子树的值,右边为右子树的值。所以我们需要扫描中序遍历序列才能找到根结点的值。
中序遍历第一个数字是根结点的值,扫描中序遍历来确定根结点所在位置。再根据中序遍历特点,根结点左边均为左子树结点的值,右边为右子树结点的值。这样我们根据前序和中序遍历找到了换份左右子树的子序列。接下来我们可以用同样的方法在对子序列构建左右子树,可以用递归来实现。
# -*- 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 not pre or not tin:
return None
root = TreeNode(pre.pop(0))
index = tin.index(root.val)
root.left = self.reConstructBinaryTree(pre, tin[:index])
root.right = self.reConstructBinaryTree(pre, tin[index+1 :])
return root