分析:对于二叉树的还原问题,首先应该对于先序遍历以及中序遍历的特点有一定的了解。每一个先序遍历的结点都是对应子树的根结点,于是在中序遍历中,位于该结点左边的为其左子树,右边的为其右子树。进行反复划分后,采取递归的方式解决问题。
class Solution {
public:
TreeNode* PreOrderBulid(vector<int>& A,vector<int>& B,int l1,int h1,int l2,int h2){
TreeNode* root=new TreeNode(A[l1]);
int i,llen,rlen;
for(i=l2;B[i]!=root->val;i++);
llen=i-l2;
rlen=h2-i;
if(llen)
root->left=PreOrderBulid(A,B,l1+1,l1+llen,l2,l2+llen-1);
else
root->left=NULL;
if(rlen)
root->right=PreOrderBulid(A,B,h1-rlen+1,h1,h2-rlen+1,h2);
else
root->right=NULL;
return root;
}
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
if(preorder.size()==0)
return {};
int l1=0,l2=0,h1=preorder.size()-1,h2=preorder.size()-1;
TreeNode* root=PreOrderBulid(preorder,inorder,l1,h1,l2,h2);
return root;
}
};