解题思路:
前序遍历是根左右,中序遍历是左根右,所以遍历时需要在前序字符串里找到根节点,然后在中序字符串里遍历到该根节点,最后对左右节点采用递归即可
代码
class Solution {
public:
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
return build(preorder ,inorder ,0,0,inorder.size()-1);
//前序 : 根左右
//中序 : 左根右
//后序 : 左右根
}
TreeNode* build(vector<int>& preorder , vector<int>&inorder, int root, int left, int right){
if(left > right){
return NULL;
}
TreeNode *topnode = new TreeNode(preorder[root]);
int i =left;
while(i <right && preorder[root] != inorder[i]){
i++;
}
topnode->left = build(preorder , inorder , root+1 , left, i-1);
topnode->right = build(preorder, inorder, root+1+i-left , i+1, right);
return topnode;
}
};