【Python学习-二叉树-递归】【剑指offer】之重建二叉树

【Python学习-二叉树-递归】【剑指offer】之重建二叉树

题目

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

基础知识

只要学懂了链表,二叉树并不难理解,链表只有一个指向,二叉树有左右两个指向,关于前序、中序、后续遍历顺序网上有很多介绍,可以仔细看看,这里就不放链接了,自己动手丰衣足食,只需要了解的是只要中序和(前序或后序)即可推导出整个二叉树序列,这个是必须要掌握的。
前序:根 >左 >右
中序:左 >根 >右
后序:左 >右 >根
为什么一定要已知中序呢?
因为当知道前序或者后序时,是可以找到根的,找到根后,那么在中序中,根在中间,正好可以将整个序列分为左右部分,左子树,右子树。下面就以题目中的例子来说明,并引申递归思想来解答该题。

例子

前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}
思想很简单:找根(在前序中找),分左右(在中序中分)
下面用该思想一步一步重建该二叉树:
在这里插入图片描述

思路分析

当输入一个前序pre 和中序tin时,首先需要在前序中找到第一个根,然后根据中序中根的位置将中序分为左边部分和右边部分,这连个部分又分别去根据前序找根,根据中序分左右,这是一个重复的过程,那么是可以利用递归来实现的,那么何时停止?需要给递归一个出口,当我们分出的两个部分的长度为1或者0时就结束了。

递归代码

# -*- 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 len(pre)==0: # 直到传入的前序长度为0 说明没有根节点了,那么值为None
           return None
       if len(pre)==1: # 直到传入的前序长度为1,说明是最后一个根节点,返回它的值即可
           return TreeNode(pre[0])
       root=TreeNode(pre[0]) #根节点
       tinL=tin[:tin.index(pre[0])] #根据根节点在中序分左右
       tinR=tin[tin.index(pre[0])+1:] #根据根节点在中序分左右
       root.left=self.reConstructBinaryTree(pre[1:tin.index(pre[0])+1],tinL) #根节点左边连接分出来的左边部分
       root.right=self.reConstructBinaryTree(pre[tin.index(pre[0])+1:],tinR) #根节点右边连接分出来的右边部分 
      #只要两个部分长度不为0或1,会一直递归
       return root
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值