https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/
解法一、递归
前序遍历的序列组成形式为:根, {左子树前序结果}, {右子树前序结果}
中序遍历的序列组成形式为:{左子树中序结果}, 根, {右子树中序结果}
把上面这两行写出来,递归解法就很容易想到了。
很明显,我们需要知道左子树的节点个数,才好获取到左子树的前序结果。由于树中无重复元素,我们可以利用中序遍历的结果,获取根的下标,这样也就间接得到了左子树中序结果的数组长度,同时也达到了我们的目的。
如果在递归过程中每次都遍历中序结果去找根的下标,会耗费太多时间。注意到,中序结果是不会变的,所以完全可以利用一个哈希表来存储 值 → \rightarrow →下标 的映射。
代码
class Solution {
public:
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
unordered_map<int, int> m;
for (int i = 0; i < inorder.size(); ++i) m[inorder[i]] = i;
return helper(m, preorder, 0, preorder.size() - 1, 0, preorder.size() - 1);
}
private:
TreeNode* helper(unordered_map