[LeetCode] 105. Construct Binary Tree from Preorder and Inorder Traversal

原题链接：https://leetcode.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/

1. 题目介绍

Given preorder and inorder traversal of a tree, construct the binary tree.
给出一棵树的前序遍历和中序遍历，根据这两种遍历顺序构造一棵二叉树

Note:
You may assume that duplicates do not exist in the tree.
你可以假设，树中没有重复的节点

For example, given
比如，给出下面的遍历序列
preorder = [3,9,20,15,7]
inorder = [9,3,15,20,7]

Return the following binary tree:
返回下面的二叉树

2. 解题思路

方法1： 递归

1. 在先序遍历中找到根节点
   先序遍历的第一个数必定是根节点，因此我们可以轻松知道根节点是谁。

2. 在中序遍历中找到根节点
   由题目可知，树中没有重复的节点，于是我们可以在中序遍历中找到根节点。

3. 在中序遍历中找到左右子树的区间
   中序遍历中，根节点之前的部分就是左子树，根节点之后的树就是右子树。同时可以计算出左子树的节点个数。

4. 在先序遍历中找到左右子树的区间
   根据左子树的长度，我们回到先序遍历，又可以找出在先序遍历中，左子树和右子树的区间在哪里。

5. 递归重复上述4步，每一步都返回当前的根节点和左右子树。左右子树交给下一层递归完成计算。

实现代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        return helper(preorder, 0, preorder.length-1, inorder, 0, inorder.length-1);
    }
    
    public TreeNode helper(int[] preorder, int p_start, int p_end, int[] inorder , int i_start, int i_end){
        if(p_start > p_end || i_start > i_end){
            return null;
        }
        int rootVal = preorder[p_start];//根节点是先序遍历的第一个节点，rootVal是根节点的值
        int rootIndex = 0;//rootIndex是根节点在中序遍历数组中的下标
        
        //在中序遍历数组中找到根节点的下标
        for(int i = i_start ; i <= i_end ; i++ ){
            if(inorder[i] == rootVal){
                rootIndex = i;
                break;
            }
        }
        
        //在中序遍历的数组中，[i_start, rootIndex-1]之间部分是左子树，
        //[rootIndex+1, i_end]之间的是右子树
        
        //在先序遍历的数组中，[p_start+1 , p_start+length]之间部分是左子树，
        //[p_start+length+1 , p_end]之间的是右子树
        
        //length是左子树的节点个数
        int length = rootIndex - i_start;
        
        TreeNode temp = new TreeNode(rootVal);
        temp.left  = helper(preorder, p_start+1, p_start+length, inorder, i_start, rootIndex-1);
        temp.right = helper(preorder, p_start+length+1, p_end, inorder, rootIndex+1, i_end);
        
        return temp;
    }
}

方法2 改进方法1，使用HashMap

主要的步骤和方法1都是一样的，不同之处在于使用了HashMap存储了中序遍历的节点值和下标。
这样做的好处是使得方法1的第二步，在中序遍历中找到根节点的下标，更加方便。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    int pre_index = 0;
    int [] pre;
    int [] in;
    HashMap<Integer, Integer> map = new HashMap<>();
    
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        pre = preorder;
        in = inorder;
        
        //将中序遍历的数组和对应的下标全都放入HashMap
        int in_index = 0;
        for(int i : inorder){
            map.put(i, in_index ++);
        }
        return helper( 0 , in.length-1);
    }
    
    TreeNode helper(int l, int r){
        if(l > r){
            return null;
        }
        
        int rootVal = pre[pre_index];//在先序遍历中找到根节点的值
        pre_index++;        
        TreeNode root = new TreeNode(rootVal);
        
        int rootIndex = map.get(rootVal);//在中序遍历中找到根节点的下标
        root.left  = helper(l , rootIndex-1);//[l, rootIndex]是左子树在中序遍历数组中的区间
        root.right = helper(rootIndex+1 , r);//[rootIndex , r]是右子树在中序遍历数组中的区间
        return root;
    }
}