根据一棵树的前序遍历与中序遍历构造二叉树

一、题目

根据一棵树的前序遍历与中序遍历构造二叉树。

注意:你可以假设树中没有重复的元素。
LeetCode题目链接

二、思路

我们知道，前序遍历的第一个元素是根节点，用根节点在中序遍历的结果数组中找到该元素下标pos，就可以将中序遍历的结果数组分成两个部分 [left,pos) 和 [pos+1,right)。在这两部分中不断递归就能够构造出一棵二叉树。

注意：除了没有左右子树，停止该部分子树的构造，若先序遍历结果的数组遍历结束，同样停止继续构造该树。

三、代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    private int index = 0; // 当前访问到先序遍历中的第几个元素
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        return buildTreeHelper(preorder,inorder,0,inorder.length);
    }
    
     /* 
            传入参数的意义：
                preorder 前序遍历结果数组
                inorder 中序遍历结果数组
                left 中序遍历时的开始位置
                right 中序遍历时的结束位置
                
     */
    private TreeNode buildTreeHelper(int[] preorder, int[] inorder, int left, int right) {
        if (left >= right){
            // 中序遍历结果为空，这个树就是空树
            return null;
        }
        if (index > preorder.length){
            // 遍历元素结束
            return null;
        }
        TreeNode root = new TreeNode(preorder[index]);
        index++;
        // 根据该节点在中序遍历结果中 的位置，把inorder数组分成左右子树两部分
        int pos = findPos(inorder,left,right,root.val);
        root.left = buildTreeHelper(preorder, inorder, left, pos);
        root.right = buildTreeHelper(preorder, inorder, pos + 1, right);
        return root;
    }

    private int findPos(int[] inorder, int left, int right, int toFind) {
        // inorder 是中序遍历的的结果数组
        // left 是中序遍历的开始位置
        // right 是中序遍历的结束位置
        // toFind 是前序遍历的第一个节点，即根节点
        for (int i = left; i < right; i++) {
            if (inorder[i] == toFind){
                return i;
            }
        }
        return -1;
    }
}