二叉树重建

一、题目描述

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

例如，给出

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

返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

题目来源：二叉树重建

二、解题思路

二叉树前序遍历的顺序为[根节点|左子树|右子树]

二叉数中序遍历的顺序为[左子树|根节点|右子树]

由此可知：

前序遍历的首个元素即为根节点

根据前序的根节点，把中序遍历划分为[左子树|根|右子树]

根据中序遍历左（右）子树节点的数量，可以将前序遍历划分为[根节点|左子树|右子树]

确定了三个节点的关系，根节点，左子树的根，右子树的根

由二叉树的定义，本题可以采用递归的方式来解决。

递归的参数为：前序遍历的根节点索引 rootIndex，中序遍历的左边界inorderLeft，中序遍历的右边界inorderRight

递归结束条件：inorderLeft>inorderRight

递归过程：

       1、先创建根节点 

       2、得到根节点在中序遍历中的索引 i

       3、构建根节点的左子树和右子树

                左子树的根为rootIndex+1，中序遍历的左边界为inorderLeft，中序遍历的右边界i-1

                右子树的根为rootIndex+i-inordereft+1,中序遍历左边界 i+1，中序遍历右边界inorderRight

代码如下：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
 //重建二叉树  递归的方式
 import java.util.*;
class Solution {
    private int []preorder;
    private Map<Integer,Integer> dict=new HashMap<Integer,Integer>();
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        this.preorder=preorder;
        for (int i=0;i<inorder.length;i++){
            dict.put(inorder[i],i);
        }
        return helper(0,0,inorder.length-1);
    }

    private TreeNode helper(int rootIndex,int inorderLeft,int inorderRight){
        if(inorderLeft>inorderRight){
            return null;
        }
        TreeNode root=new TreeNode(preorder[rootIndex]);
        int i=dict.get(preorder[rootIndex]);
        root.left=helper(rootIndex+1,inorderLeft,i-1);
        root.right=helper(rootIndex+i-inorderLeft+1,i+1,inorderRight);
        return root;
    }
}