剑指offer-Python-04-重建二叉树

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值