标签:root 一个 return 相等 基本 思路 sign node tree
根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树:
3
/ 9 20
/ 15 7
思路: 根据构造二叉树的流程,中序遍历的访问顺序为左-中-右;后序遍历的方位顺序为左-右-中。
TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder)
{
int n=inorder.size();
if(n>0)
{
TreeNode* root=new TreeNode(postorder[n-1]);
vector<int>::iterator it=find(inorder.begin(),inorder.end(),postorder[n-1]);
vector<int> il,ir,pl,pr;
il.assign(inorder.begin(),it);
ir.assign(it+1,inorder.end());
int l,r;
l=il.size();
r=ir.size();//中序后序遍历长度相等
pl.assign(postorder.begin(),postorder.begin()+l);
pr.assign(postorder.begin()+l,postorder.end());
root->left=buildTree(il,pl);
root->right=buildTree(ir,pr);
return root;
}
else return NULL;
}
根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/ 9 20
/ 15 7
思路:
和中序后序构造思路基本一致
根据构造二叉树的流程,前序遍历的访问顺序为中-左-右,中序遍历的访问顺序为左-中-右。
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder)
{
int n=preorder.size();
if(n>0)
{
TreeNode* root=new TreeNode(preorder[0]);
vector<int>::iterator it=find(inorder.begin(),inorder.end(),preorder[0]);
vector<int> il,ir,pl,pr;
il.assign(inorder.begin(),it);
ir.assign(it+1,inorder.end());
int l,r;
l=il.size();
r=ir.size();//中序前序遍历长度相等
pl.assign(preorder.begin()+1,preorder.begin()+l+1);
pr.assign(preorder.begin()+l+1,preorder.end());
root->left=buildTree(pl,il);
root->right=buildTree(pr,ir);
return root;
}
else return NULL;
}
标签:root 一个 return 相等 基本 思路 sign node tree
原文地址:https://www.cnblogs.com/wendyy/p/9332633.html