题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
复习知识点:二叉树的遍历
前中后序指的是根节点的顺序,左总在右的前面
前序遍历的顺序是:根节点,左子节点(左子树),右子节点(右子树)。
中序遍历的顺序是:左子节点,根节点,右子节点。
后序遍历的顺序是:左子节点,右子节点,根节点。
这三种遍历都有递归和循环两种不同的实现方式,每种遍历的递归实现都比循环实现简洁很多。
由遍历顺序重构二叉树:其实就是把前序遍历中得到的根节点在中序遍历中找到就可以分出左右子树,不断重复这个过程
Python题解:
# -*- 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
return self.findSubTree(pre,tin)
def findSubTree(self,front,mid):
if len(front)==0:
#print('null')
return None
if len(front)==1:
#print('return value '+str(front[0]))
return TreeNode(front[0])
else:
r = TreeNode(front[0])
index = mid.index(front[0])
#print('root:',front[0])
#print('寻找 '+str(front[0])+' 的左子树')
r.left=self.findSubTree(front[1:index+1],mid[0:index])
#print('寻找 '+str(front[0])+' 的右子树')
r.right=self.findSubTree(front[index+1:],mid[index+1:])
#print('子树 '+str(front[0])+' 结束')
return r
以前序遍历[1,2,4,7,3,5,6,8]和中序遍历[4,7,2,1,5,3,8,6]为例,上述代码中的print取消注释,得到的重构二叉树过程如下:
('root:', 1)
寻找 1 的左子树
('root:', 2)
寻找 2 的左子树
('root:', 4)
寻找 4 的左子树
null
寻找 4 的右子树
return value 7
子树 4 结束
寻找 2 的右子树
null
子树 2 结束
寻找 1 的右子树
('root:', 3)
寻找 3 的左子树
return value 5
寻找 3 的右子树
('root:', 6)
寻找 6 的左子树
return value 8
寻找 6 的右子树
null
子树 6 结束
子树 3 结束
子树 1 结束