【Leetcode】105. Construct Binary Tree from Preorder and Inorder Traversal

题目地址：

https://leetcode.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/

给定一棵树的先序和中序遍历，返回这棵树。题目保证这棵树的每个数两两不同。

思路是用递归。由先序遍历可以确定这棵树的根，然后再在中序遍历里去找这个树根，这样在中序遍历的序列里，树根左边就是左子树的所有节点，右边就是右子树的所有节点，也知道了左右子树的size是多少。这样就可以再在先序遍历的树根后数若干数作为左子树，再数若干数作为右子树，再递归求解即可。为了更快的在中序遍历里找到树根，可以用一个哈希表记录每个数在中序遍历数组的下标。代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left),
 * right(right) {}
 * };
 */
class Solution {
 public:
  TreeNode* buildTree(vector<int>& pre, vector<int>& in) {
    unordered_map<int, int> mp;
    int n = in.size();
    for (int i = 0; i < n; i++) mp[in[i]] = i;
    return dfs(pre, 0, n - 1, in, 0, n - 1, mp);
  }

  TreeNode* dfs(auto& pre, int b1, int e1, auto& in, int b2, int e2, auto& mp) {
    if (b1 > e1) return nullptr;
    auto root = new TreeNode(pre[b1]);
    int idx = mp[root->val], sz_left = idx - b2;
    root->left = dfs(pre, b1 + 1, b1 + sz_left, in, b2, idx - 1, mp);
    root->right = dfs(pre, b1 + sz_left + 1, e1, in, idx + 1, e2, mp);
    return root;
  }
};

时空复杂度$O(n)$。

注解：
关于先序遍历和中序遍历可以确定一棵二叉树的结论，可以用数学归纳法证明。设树节点个数为$0$或$1$，则结论成立。假设对于节点个数小于等于$n$的树结论成立。当节点个数为$n+1$时，由上面的算法可以先找到树根，接着由于左右子树节点数严格小于$n+1$，由归纳假设，左右子树唯一确定，所以整棵树也唯一确定。