剑指Offer-重建二叉树

题目描述

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

解题思想：递归思想！
使用前序和中序遍历可以确认唯一的二叉树，通过前序可以得知根节点为1，然后通过中序可以得知左子树为{4, 7, 2}，右子树为{5, 3, 8, 6}。接下来通过递归思想，{4, 7, 2}中4为根节点，再继续分左右子树，对{5, 3, 8, 6}也做同样操作，直至遍历结束。

虽然很容易能想到递归求解，但是开始写代码时可能会卡住，不知从何下手，需要传递多少参数，本题最主要就是要确定左子树与右子树在数组间的范围，再去判断下一步谁是根谁是左谁是右。
或者可以使用Arrays.copyOfRange复制数组，将左子树与右子树单独复制出来进行递归。

Java最优解

	public class TreeNode {
         int val;
         TreeNode left;
         TreeNode right;
         TreeNode(int x) { val = x; }
     }
    public TreeNode findBinaryTree(int[] pre, int preStart, int preEnd, int[] in ,int inStart, int inEnd){
        // 没有符合前序和中序的二叉树或已经到达叶子节点
        if(preStart>preEnd || inStart>inEnd)
            return null;

        // 根节点
        TreeNode root = new TreeNode(pre[preStart]);

        for(int i=inStart;i<=inEnd;i++){
            if(in[i]==pre[preStart]){
                // i-startIn 为左子树个数
                root.left = findBinaryTree(pre, preStart+1, i-inStart+preStart, in, inStart, i-1);
                // (i-startIn+startPre)+1：左子树的下一个开始为右子树
                root.right = findBinaryTree(pre, i-inStart+preStart+1, preEnd, in, i+1, inEnd);
            }
        }
        return root;
    }
	// 重建主函数
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        TreeNode root = findBinaryTree(pre, 0, pre.length-1, in, 0, in.length-1);
        return root;
    }

使用Arrays.copyOfRange

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