题目描述:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
解题思路:
(1)根据前序遍历结果,第一个元素为二叉树的根结点;
(2)观察中序遍历结果,根结点左侧的为左子树,若左子树根结点前(后)再无任何元素,则左(右)子树的左分支为空;根结点右侧的为右子树,若右子树根结点前(后)再无任何元素,则左(右)子树的左分支为空;
即上面的过程是递归的。先找到当前树的根结点,然后划分为左右子树,再进入左子树重复上面的过程,最后进入右子树重复上面的过程,最终还原一棵树。
小虾稍微动手分析了一下,但字体不太好看嘻嘻(迁就着看,哈哈哈)
代码:(python)
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
# 返回构造的TreeNode根节点
#pre表示前序遍历的序列,tin表示中序遍历的序列
def reConstructBinaryTree(self, pre, tin):
if not pre or not tin:#先判断一下,前序和中序是否为空
return None
root = TreeNode(pre[0]) #前序遍历的第一个表示root结点
rootIndex = tin.index(pre[0])#root在中序遍历序列中的索引
#根据中序遍历,处理左子树,切片表示在前序和中序中的左子树
root.left = self.reConstructBinaryTree(pre[1:rootIndex+1], tin[0:rootIndex])
#根据中序遍历,处理右子树,切片表示在前序和中序中的右子树
root.right= self.reConstructBinaryTree(pre[rootIndex+1:], tin[rootIndex+1:])
return root