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

QUESTION

medium

题目描述

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

例如，给出

• 前序遍历 preorder = [3,9,20,15,7]

• 中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

说明

你可以假设树中没有重复的元素

---

SOLUTION

一开始这个问题确实让我头疼，但仔细思考两种遍历，你可以发现：前序遍历适合用于确定子树的根节点，根据这个根节点找它在中序遍历的位置，然后把中序遍历一分为二，左边属于左子树，右边属于右子树，这样眼前就豁然开朗了

方法一(递归)

根据前面的想法最容易想到的就是递归了吧，其实和遍历本质上是相同的，但是要注意判断条件

• 所有的前提是基于前序遍历和中序遍历匹配且正确的情况下，经测试不需要考虑不匹配的情况
• 用了一个 root 记录当前子树根节点的位置，注意值是一直在变的
• 查找方法（注释内的代码）没有优化，用的顺序查找，可以改进（在耗时较少的回答里有用Map提升查找速度的可自行测试）

TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
    int root = 0;
    return build(preorder, inorder, root, 0, inorder.size() - 1);
}
TreeNode* build(vector<int>& preorder, vector<int>& inorder, int &root, int l, int r){
    if(l > r) return NULL;
    TreeNode* node = new TreeNode(preorder[root]);
    //find preorder[root] in inorder
    int mid;
    for(mid = l; mid <= r; mid++){
        if(inorder[mid] == preorder[root]){
            break;
        }
    }
    //find end
    root++;
    node->left = build(preorder, inorder, root, l, mid - 1);
    node->right = build(preorder, inorder, root, mid + 1, r);
    return node;
}