剑指Offer . 重建二叉树

剑指Offer.重建二叉树

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

eg：  前序遍历 preorder = [3,9,20,15,7]
          中序遍历 inorder = [9,3,15,20,7]

返回如下二叉树
     3
    /  \
  9   20
  /  \
15   7

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public Map<Integer,Integer> HashMap = new HashMap<Integer,Integer>();
    
	//将中序遍历数组存到map集合中
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        for(int i = 0 ; i < preorder.length ; i++){
            HashMap.put(inorder[i],i);
        }
        return asd(preorder,inorder,0,preorder.length-1,0,inorder.length-1);
    }
	
	//先找到每个树的根结点，然后定位中序数组根结点的位置，该位置前面的元素均为根结点的左子树元素，然后递归寻找作树的根结点。右边同理。
    public TreeNode asd(int[] preorder , int[] inorder ,int pre_left,int pre_right,int ino_left,int ino_right){
    	//递归返回条件
        if(pre_left > pre_right){
            return null;
        }
		//每个前序数组的第一个值均为根节点
        TreeNode root = new TreeNode(preorder[pre_left]);
        //找到中序数组中根结点的角标，该脚本之前的元素均为根结点的左子树元素
        int left = HashMap.get(preorder[pre_left]);
        int length = left - ino_left ;
        //左右递归
        root.left=asd(preorder,inorder,pre_left+1,pre_left+length,ino_left,left-1);
        root.right=asd(preorder,inorder,pre_left+length+1,pre_right,left+1,ino_right);

        return root;
    }
}