题目描述
输入某二叉树的前序遍历和中序遍历的结果,请构建该二叉树并返回其根节点。
假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
示例 1:
Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]
示例 2:
Input: preorder = [-1], inorder = [-1]
Output: [-1]
限制:
0 <= 节点个数 <= 5000
分析
在二叉树的前序遍历序列中,第一个数字总是树的根节点的值。但是在中序遍历序列中,根节点的值在序列的中间,左子树的节点的值位于根节点的值的左边,右子树的节点的值位于根节点的值的右边。
所以我们要先开一个hash表来记录中序遍历的下标。
然后在前序遍历中找到根节点,根据hash表找到中序遍历中根节点的位置,这样就能确定左右子树的范围。然后递归构建左右子树。
代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
unordered_map<int, int> hash;
TreeNode* Build(vector<int>& preorder, vector<int>& inorder, int pre_l, int pre_r, int in_l, int in_r) {
if (pre_l > pre_r) return NULL;
int h_root = hash[preorder[pre_l]];
int left_size = h_root - in_l;
TreeNode* root = new TreeNode(inorder[h_root]);
root->left = Build(preorder, inorder, pre_l + 1, pre_l + left_size, in_l, h_root - 1);
root->right = Build(preorder, inorder, pre_l + 1 + left_size, pre_r, h_root + 1, in_r);
return root;
}
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
int n = preorder.size();
for (int i = 0; i < n; i++)
hash[inorder[i]] = i;
return Build(preorder, inorder, 0, n - 1, 0, n - 1);
}
};