思路:先根据二叉树的的前序遍历确定根节点,然后再根据中序遍历确定根节点的左右子树进行建树。
在中序数组中寻找根节点的时候需要O(n)遍历才能得到,可以先用一个哈希表存储起来,之后只要O(1)即可找到根节点在中序数组中的位置。
class Solution {
public:
/*
核心思想就是根据前序确定根节点,
然后根据根节点和中序遍历确定左右分支,
进行递归
*/
unordered_map<int, int> m;
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
for(int i = 0;i<inorder.size();i++){
m[inorder[i]] = i;
}
return build(preorder, 0, 0, inorder.size()-1);
}
/*
@ param index根节点的下标
@ param l 子树的左边界
@ param r 子树的右边界
*/
TreeNode* build(vector<int>& preorder, int index, int l, int r){
if(l > r) return NULL;
//先创建根节点
TreeNode* node = new TreeNode( preorder[index] );
//找出根节点的在中序数组中的下标
int root = m[preorder[index]];
node->left = build(preorder, index+1, l, root-1);
//前序根左右,因此右子树根节点的下标是 当前根节点+左子树节点数+1
node->right = build(preorder, index + (root - l)+1, root+1, r);
return node;
}
};