剑指offer:重建二叉树（JAVA实现最新版）

题目描述

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

代码实现及解题思路

1. 由前序遍历可得知根节点为第一位，而在中序遍历中，根节点是隔开左右孩子的，所以根节点的左边为左子树，根节点的右边为右子树；
2. 由步骤1得出左子树和右子树的长度，分别得出左右子树的前序遍历和中序遍历；
3. 重复步骤1、2，直到左右子树的长度为0，依次返回寻得的根节点。

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
         if(pre.length == 0) {
		  return null;
	  }
	TreeNode root = new TreeNode(pre[0]);//pre[0]是根
	int len = pre.length;
	if(len == 1) {
		root.left = null;
		root.right = null;
		return root;
	}
	//找到根位置index
	int rootnode = root.val;
	int i;
	for(i = 0; i<len;i++) {
		if(rootnode == in[i])
			break;
	}
if(i>0){
        
		int[] pre_left=new int[i];
		int[] in_left=new int[i];
		for(int j=0;j<i;j++){
			pre_left[j]=pre[j+1];
		}
		for(int j=0;j<i;j++){
			in_left[j]=in[j];
		}
		root.left=reConstructBinaryTree(pre_left,in_left);
	}else{
		root.left=null;
	}
	//创建右子树
	if(len-i-1>0){
		int[] pre_right=new int[len-i-1];
		int[] in_right=new int[len-i-1];
		for(int j=i+1;j<len;j++){            //j=i+1,因为i为中序根节点位置，也为前序遍历左子树的最后一个节点的位置
			pre_right[j-i-1]=pre[j];
			in_right[j-i-1]=in[j];
		}
		root.right=reConstructBinaryTree(pre_right,in_right);
	}else{
		root.right=null;
	}
	return root;
	

    }
}