从前序与中序遍历序列构造二叉树

/*
 * 解题思路：
 * 1. 通过流处理查找根节点在中序遍历中的位置。
 * 2. 使用流构建左右子树。
 * 3. 递归构建整个树。
 */
 
import java.util.*;
import java.util.stream.*;
 
class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}
 
public class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if (preorder.length == 0) return null;
        int rootVal = preorder[0];
        TreeNode root = new TreeNode(rootVal);
        int inRootIndex = IntStream.range(0, inorder.length)
                                   .filter(i -> inorder[i] == rootVal)
                                   .findFirst()
                                   .orElse(-1);
        int[] leftInorder = Arrays.copyOfRange(inorder, 0, inRootIndex);
        int[] rightInorder = Arrays.copyOfRange(inorder, inRootIndex + 1, inorder.length);
        int[] leftPreorder = Arrays.stream(preorder)
                                   .skip(1)
                                   .limit(leftInorder.length)
                                   .toArray();
        int[] rightPreorder = Arrays.stream(preorder)
                                    .skip(1 + leftInorder.length)
                                    .toArray();
        root.left = buildTree(leftPreorder, leftInorder);
        root.right = buildTree(rightPreorder, rightInorder);
        return root;
    }
}