06--重建二叉树

最新推荐文章于 2021-10-24 13:03:41 发布

大号小白兔

最新推荐文章于 2021-10-24 13:03:41 发布

阅读量408

点赞数

分类专栏：剑指offer--java版

本文链接：https://blog.csdn.net/a1b2c3d4123456/article/details/97644872

版权

剑指offer--java版专栏收录该内容

4 篇文章 0 订阅

订阅专栏

根据给定的二叉树前序遍历和中序遍历序列，可以唯一确定该二叉树。解题方法是利用前序遍历的第一个元素作为根节点，然后在中序遍历中找到根节点的位置，以此划分左右子树，并通过递归继续构建子树。递归过程中需计算左右子树的参数，最终重建出完整的二叉树结构。

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

题目：

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

解题思路：

可以通过前序遍历和中序遍历确定一棵二叉树，也可以通过后序遍历和中序遍历确定一棵二叉树，但是无法通过前序遍历和后序遍历确定一棵二叉树，因为要借助中序遍历里面的根节点获取左右子树。

前序遍历第一个数字就是根节点，中序遍历中找到该值的位置，在中序遍历中该值左侧是左子树，右侧是右子树，以此类推，递归求出所有的节点。

递归的传参是前序遍历的数组，前序遍历的开始和结束，中序遍历的数组，中序遍历的开始和结束，难点在于求出确定了根节点之后，左右节点的传参，假设第一次调用传参为（int [] pre,int preStart,int preEnd,int [] in,int inStart,int inEnd）值为（pre, 0, pre. length, in, 0, in.length），先找到根节点，curRoot=TreeNode(pre[preStart]),也就是前序遍历的第一个值，左子树的传参pre,preStart+1, preStart+i-inStart+1, in,inStart, i。因为preStart下一个就是左子树，i是遍历中序遍历数组时候等于根节点时候的下标，preStart+1+i-inStart就是preEnd，inStart就是中序遍历的开始，i就是中序遍历的结尾。向左取值，左闭右开。同理可知右子树的传参为pre,preStart+i-inStart+1,preEnd,in,i+1,inEnd。

代码

public class Solution {
  
   public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
       return reconstructBinaryTree(pre, 0, pre. length, in, 0, in.length);
    }
    
    private TreeNode reconstructBinaryTree(int [] pre,int preStart,int preEnd,int [] in,int inStart,int inEnd){
        if(inStart==inEnd) {
            return null;
        }
        TreeNode curRoot = new TreeNode(pre[preStart]);
        for(int i=0;i<inEnd;i++){
            if(in[i]==pre[preStart]){
                curRoot.left=reconstructBinaryTree(pre,preStart+1,preStart+i-inStart+1,in,inStart,i);
                curRoot.right=reconstructBinaryTree(pre,preStart+i-inStart+1,preEnd,in,i+1,inEnd);
                break;
            }
        }
        return curRoot;
    }
}