剑指offer-7 重建二叉树
一、题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
二、解题
1.解题思路
前序遍历:根→左→右
中序遍历:左→根→右
- 由前序遍历序列pre={1,2,4,7,3,5,6,8}可知根结点是1;
- 则在中序遍历序列in={4,7,2,1,5,3,8,6}中找到1,便可知1所在位置的左侧是左子树,1所在位置的右侧是右子树;
- 递归调用:将左子树和右子树分别看成一颗树,将其前序遍历序列、中序遍历序列分别传入到该方法中,便可得到左子树的根结点、右子树的根结点。此时需要用第一步得到的根结点连接它们;
- 递归调用的终止条件:直到传入数组为空,说明已经没有节点,直接返回null。
2.代码实现
# -*- 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
#TreeNode构建树,树的根节点为prt[0]
root=TreeNode(pre[0])
#在中序遍历中找到根节点的位置
index=tin.index(pre[0])
#开始递归,index左侧的为左子树,index右侧的为右子树
#root.left=self.reConstructBinaryTree(pre, tin)
root.left=self.reConstructBinaryTree(pre[1:index+1], tin[:index])
root.right=self.reConstructBinaryTree(pre[index+1:], tin[index+1:])
return root
3.总结
对于二叉树的重新构建,由于给定了树的前序以及中序遍历的结果,前序遍历的第一个数据即为二叉树的根节点,找到根节点之后,再去中序遍历中寻找根节点的索引,在索引之前的为左子树,索引之后的为右子树。