题目地址:https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/
    难度:中等
    思路:
    首先需要了解前序和中序的定义:

    • 前序序列即:根 -> 左 -> 右
    • 中序序列即:左 -> 根 -> 右

    因此,可以通过前序序列得知要构造的二叉树的根节点即其第一个节点, 而如果在给定的中序序列数组中找到这个根节点,就可以确定其左右子树的分界点。
    image.png

    1. public class Soultion {
    2. public TreeNode buildTree(int[] preorder, int inorder) {
    3. // 保存给定的前序和中序数组长度
    4. int preLen = preorder.length;
    5. int inLen = inorder.length;
    6. // 异常检查(可选操作)
    7. if(preLen != inLen) {
    8. throw new RuntimeException("Incorrect input data");
    9. }
    10. Map<Integer, Integer> map = new HashMap<>();
    11. for(int i = 0; i < preLen; i++) {
    12. map.put(inorder[i], i);
    13. }
    14. return buildTreeHelper(preorder, 0, preLen - 1, map, 0, inLen - 1);
    15. }
    16. private TreeNode buildTreeHelper(int[] preorder, int preLeft, int preRight, Map<Integer, Integer> map, int inLeft, int inRight) {
    17. // 递归函数的终止条件
    18. if (preLeft > preRight || inLeft > inRight) {
    19. return null;
    20. }
    21. // 根据前序遍历获取根节点的值
    22. int rootVal = preorder[preLeft];
    23. TreeNode root = new TreeNode(rootVal);
    24. // 中序遍历中,根节点的下标
    25. int pIndex = map.get(rootVal);
    26. root.left = buildTreeHelper(preorder, preLeft + 1, pIndex - inLeft + preLeft, map, inLeft, pIndex - 1);
    27. root.right = buildTreeHelper(preorder, pIndex - inLeft + preLeft + 1, preRight, map, pIndex + 1, inRight);
    28. return root;
    29. }
    30. }