剑指offer 4. 重建二叉树

 1 import java.util.Arrays;
 2 public class Solution {
 3     // 递归将中序遍历和前序遍历数组变成分成子树的 前序 遍历和 中序遍历的数组
 4     public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
 5         if(pre.length == 0 || in.length == 0){
 6             return null;
 7         }
 8         // 创建根节点
 9         TreeNode root = new TreeNode(pre[0]);
10         
11         // 从中序遍历中找到根节点，
12         for(int i = 0; i < in.length; i++){
13             if(in[i] == pre[0]){
14                 // 左子树，注意copyOfRange的区间范围，左闭右开
15                 root.left = reConstructBinaryTree(Arrays.copyOfRange(pre, 1, i + 1), Arrays.copyOfRange(in, 0, i));
16                 // 右子树
17                 root.right = reConstructBinaryTree(Arrays.copyOfRange(pre, i + 1, pre.length), Arrays.copyOfRange(in, i + 1, in.length));
18             }
19         }
20         return root;
21     }
22 }