剑指offer-Java实现:题目4、重建二叉树

题目描述

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

思路：了解二叉树的前，中，后序遍历的规律，我们就可以知道，通过前 中，或者 后 中 两种都能得到二叉树。主要通过三个步骤：

1.查看前（后）序遍历序列，最开始（后面）的必然是根节点。

2.获得根节点后，查看中序遍历序列，根节点左右两边分别为左子树的所有节点，和右子树的所有节点  4 7 2  ，5 3 8 6.

3.分别对两边序列重复 1 . 2 两个步骤。最终能得到二叉树。

代码：

import java.util.Arrays;
public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        if (pre == null || in == null) {
            return null;
        }
        if (pre.length == 0 || in.length == 0) {
            return null;
        }
        if (pre.length != in.length) {
            return null;
        }
        TreeNode root = new TreeNode(pre[0]);
        for (int i = 0; i < pre.length; i++) {
            if (pre[0] == in[i]) {
                root.left = reConstructBinaryTree(Arrays.copyOfRange(pre,1,i+1),
                                              Arrays.copyOfRange(in,0,i));
                root.right = reConstructBinaryTree(Arrays.copyOfRange(pre,i+1,pre.length),
                                               Arrays.copyOfRange(in,i+1,in.length));
            }
        }
        return root;
    }

}