剑指offer31 — 重建二叉树
题目
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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 or len(pre)!= len(tin):
return None
# 1.找到root以及其在中序中的位置
root = pre[0]
rootNode = TreeNode(root)
pos = tin.index(root)
# 2.新建列表,存储root的左右子树的前序和中序
tinLeft = tin[:pos]
tinRight = tin[pos+1:]
preLeft = pre[1:pos+1]
preRight = pre[pos+1:]
# 3. 递归;左右子树分别作为一个新树开始找其根结点以及左右子树
leftNode = self.reConstructBinaryTree(preLeft, tinLeft)
rightNode = self.reConstructBinaryTree(preRight, tinRight)
# 判断leftNode和rightNode是否为空,若不为空,则建立其和root的关系
if leftNode:
rootNode.left = leftNode
if rightNode:
rootNode.right = rightNode
return rootNode