《剑指Offer》Java实现-重建二叉树

题目描述

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

思路

前序遍历的第一个节点，肯定是根节点。找到在中序遍历中的根节点位置。左面的值就是左子树，右面的值就是右子树。根据这个可以进行递归。就能重建整个二叉树。

代码

    public class Solution {
        public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
           if(pre == null || in == null){
                return null;
            }
            TreeNode mm = reConstructBinaryTreeCore(pre, in, 0, pre.length-1, 0, in.length-1);
            return mm;
        }

         public  TreeNode reConstructBinaryTreeCore(int[] pre, int[] in, int preStart, int preEnd, int inStart, int inEnd) {
            TreeNode tree = new TreeNode(pre[preStart]);
            tree.left = null;
            tree.right = null;
            if (preStart == preEnd && inStart == inEnd) {
                return tree;
            }
            int root = 0;
            for(root= inStart; root < inEnd; root++){
                if (pre[preStart] == in[root]) {
                    break;
                }
            }
            int leifLength = root - inStart;
            int rightLength = inEnd - root;
            if (leifLength > 0) {
                tree.left = reConstructBinaryTreeCore(pre, in, preStart+1, preStart+leifLength, inStart, root-1);
            }
            if (rightLength > 0) {
                tree.right = reConstructBinaryTreeCore(pre, in, preStart+1+leifLength, preEnd, root+1, inEnd);
            }
            return tree;
        }
    }