【剑指Offer】4、重建二叉树

最新推荐文章于 2024-10-16 10:35:43 发布

白露塞纳

最新推荐文章于 2024-10-16 10:35:43 发布

阅读量207

点赞数

文章标签：重建二叉树二叉树数据结构先序中序

本文链接：https://blog.csdn.net/HeavenDan/article/details/103593405

版权

根据给定的前序遍历和中序遍历序列，这篇博客介绍了如何重建二叉树。博主分享了递归实现的代码，并通过例子和变量输出详细解释了重建过程，加深了对二叉树遍历和递归理解。

摘要由CSDN通过智能技术生成

一、题目描述

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

二、代码

思路：
顺序读前序遍历序列，第一个节点为根节点，再在中序遍历序列中以根结点为分割点将其分为左右子树。依此逻辑，向下递归。

举例：

# Input
Pre arr: 1 2 4 7 3 5 6 8
In  arr: 4 7 2 1 5 3 8 6

# 注：LT：Left Tree； RT：Right Tree
# 1st step:确定根节点
Pre arr: 1 2 4 7 3 5 6 8
         ▲
         1为根节点
In  arr: 4 7 2    1    5 3 8 6
        {
   4 7 2}  [1]  {
   5 3 8 6}
         {
   LT}   [root]  {
   RT}

# 2nd step:递归处理左子树
Pre arr: 2 4 7
         ▲
         2为根节点
In  arr: 4 7   2
        {
   4 7} [2]
        {
   LT}  [root]{
   NULL}
……（省略再次递归若干步）

# 3rd step:递归处理右子树
Pre arr: 3     5   6 8
In  arr: 5     3   8 6
        {
   LT}[Root]{
   RT}
……（省略再次递归若干步）

代码（参考欲风）：

public class Solution {
   
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
   
        TreeNode root=reConstructBinaryTree(pre,0,pre.length-1,in,0,in.length-1);
        return root;
    }
    //前序遍历{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}
    private TreeNode reConstructBinaryTree(int [] pre,int startPre,int endPre,int [] in,int startIn,int endIn) {
   
         
        if(startPre>endPre||startIn>endIn)
            return null;
        TreeNode root=new TreeNode(pre[startPre]);
         
        for(int i=startIn;i<=endIn;i++)
            if(in[i]==pre[startPre]){
   
                root.left=reConstructBinaryTree(pre,startPre+1,startPre+i-startIn,in,startIn,i-1);
                root.right=reConstructBinaryTree(