思路
- 前序遍历序列确定根结点——开头第一个元素
- 中序遍历序列 定位根结点——建立哈希加快查找
- 获得中序遍历的分割点,也获取了左子树的个数——用于分割的位置
- 根据获得的信息进行左右子树的序列区域分割 和 建立左右子树
代码
class Solution {
private:
unordered_map<int, int> index;
public:
TreeNode* myBuildTree(const vector<int>& preorder, const vector<int>& inorder, int preorder_left, int preorder_right, int inorder_left, int inorder_right) {
if (preorder_left > preorder_right)
return nullptr;
int preorder_root = preorder_left;
int inorder_root = index[preorder[preorder_root]];
int size_left_subtree = inorder_root - inorder_left;
TreeNode* root = new TreeNode(preorder[preorder_root]);
root->left = myBuildTree(preorder, inorder, preorder_left + 1, preorder_left + size_left_subtree, inorder_left, inorder_root - 1);
root->right = myBuildTree(preorder, inorder, preorder_left + size_left_subtree + 1, preorder_right, inorder_root + 1, inorder_right);
return root;
}
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
int n=preorder.size();
for (int i = 0; i < preorder.size(); ++i)
index[inorder[i]] = i;
return myBuildTree(preorder, inorder, 0, n - 1, 0, n - 1);
}
};