Java实现根据二叉树的前序遍历和中序遍历序列重建二叉树（递归实现）

题目描述

题目来源：牛客-剑指offer

题目描述

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

分析

根据前序遍历和中序遍历的特点，前序遍历的第一个元素即根节点，在中序序列中该节点的左边为左子树，右边为右子树。

Java实现

	public TreeNode reConstructBinaryTree(int[] pre, int[] in) {

      return reConstructBinaryTree(pre, 0, pre.length - 1 , in, 0, in.length - 1 );
    }
    private static TreeNode reConstructBinaryTree(int[] pre , int preLow, int preHigh, int[] in, int inLow,int inHigh){

        if(preLow > preHigh || inLow > inHigh) return null;
        TreeNode treeNode = new TreeNode(pre[preLow]);           //根节点为pre的第一个元素
        for (int i = 0; i < in.length; i++) {
            //找到pre[lowHigh]在in序列中的索引（即根节点在in序列中位置，根节点左边为左子树节点，右边为右子树节点）
            if( pre[preLow] == in[i] ){//startPre+i-startIn  i-startIn+startPre+1
                treeNode.left = reConstructBinaryTree(pre, preLow + 1 , preLow + i - inLow, in, inLow , i - 1);
                treeNode.right = reConstructBinaryTree(pre, i - inLow + preLow + 1, preHigh, in, i + 1, inHigh);
                break;
            }
        }
        return treeNode;
    }