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

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

给定两个整数数组 preorder 和 inorder ，其中 preorder 是二叉树的先序遍历， inorder 是同一棵树的中序遍历，请构造二叉树并返回其根节点。

输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
输出: [3,9,20,null,null,15,7]
示例 2:

输入: preorder = [-1], inorder = [-1]
输出: [-1]
解题思路：
对于前序和中序的遍历，第一步是找到根节点，然后将根节点左右分开，代表左子树和右子树，其中左子树的区间范围在中序序列中可以找到，右子树的区间范围可以在中序遍历的范围中找到，重点是在中序序列中找到根节点的位置，然后进行递归即可
第一步：在前序序列中找到根节点
第二步：在中序序列中找到根节点的位置，然后将中序序列根据根节点分成左右两个序列分别代表左右子树，
第三步：递归调用左子树和右子树，从而生成二叉树
第四步：找到递归的结束条件，当递归的当前节点为空时，返回空节点

class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        // preorder: [root, [left], [right]]
        // inorder:  [[left], root, [right]]
        return buildTree(preorder, 0, preorder.size() - 1, inorder, 0, inorder.size() - 1);
    }
    TreeNode* buildTree(vector<int>& preorder, int lo1, int hi1, vector<int>& inorder, int lo2, int hi2) {
        if (lo1 > hi1 || lo2 > hi2) return nullptr;
        int root = preorder[lo1];
        int mid = lo2;
        // 在 inorder 中查找 root 位置
        for (int i = lo2; i <= hi2; ++i) {
            if (inorder[i] == root) {
                mid = i;
                break;
            }
        }
        
        auto s = new TreeNode(root);
        // 下面的数组表示分割长度
        // inorder: [mid-lo2, mid, hi2-mid]
        // preorder:[root, mid-lo2, hi2-mid]
        s->left = buildTree(preorder, lo1+1, lo1+mid-lo2, inorder, lo2, mid-1);
        s->right = buildTree(preorder, lo1+mid-lo2+1, hi1, inorder, mid+1, hi2);
        return s;
    }
};