根据前序遍历和中序遍历树构造二叉树.
描述
根据前序遍历和中序遍历树构造二叉树.
你可以假设树中不存在相同数值的节点
您在真实的面试中是否遇到过这个题?
是
样例
给出中序遍历:[1,2,3]
和前序遍历:[2,1,3]
. 返回如下的树:
2
/ \
1 3
class Solution {
public:
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
return construct(pre,0,pre.size()-1,vin,0,vin.size()-1);
}
TreeNode* construct(vector<int>&pre,int pleft,int pright,vector<int>&vin,int vleft,int vright)
{
if(pleft>pright||vleft>vright) return NULL;
TreeNode* cur=new TreeNode(pre[pleft]);
int i=0;
for(i=vleft;i<vin.size();i++)
{
if(cur->val==vin[i])
break;
}
cur->left=construct(pre,pleft+1,i+pleft-vleft,vin,vleft,i-1);
cur->right=construct(pre,i+pleft-vleft+1,pright,vin,i+1,vright);
return cur;
}
};
根据中序遍历和后序遍历重建二叉树
样例
给出树的中序遍历: [1,2,3] 和后序遍历: [1,3,2]
返回如下的树:
2
/ \
1 3
递归解法(分治法)
分析:
这道题要求从中序和后序遍历的结果来重建原二叉树,我们知道中序的遍历顺序是左-根-右,后序的顺序是左-右-根,对于这种树的重建一般都是采用递归来做,可参见我之前的一篇博客Convert Sorted Array to Binary Search Tree 将有序数组转为二叉搜索树。针对这道题,由于后序的顺序的最后一个肯定是根,所以原二叉树的根节点可以知道,题目中给了一个很关键的条件就是树中没有相同元素,有了这个条件我们就可以在中序遍历中也定位出根节点的位置,并以根节点的位置将中序遍历拆分为左右两个部分,分别对其递归调用原函数。代码如下:
class Solution {
public:
/**
* @param inorder: A list of integers that inorder traversal of a tree
* @param postorder: A list of integers that postorder traversal of a tree
* @return: Root of a tree
*/
TreeNode * buildTree(vector<int> &inorder, vector<int> &postorder) {
// write your code here
return construct(inorder,0,inorder.size()-1,postorder,0,postorder.size()-1);
}
TreeNode* construct(vector<int>&inorder,int ileft,int iright,vector<int>&postorder,int pleft,int pright)
{
if(ileft>iright||pleft>pright)
return NULL;
TreeNode* cur=new TreeNode(postorder[pright]);
int i=0;
for(i=ileft;i<inorder.size();i++)
{
if(inorder[i]==cur->val)
break;
}
cur->left=construct(inorder,ileft,i-1,postorder,pleft,pleft+i-1-ileft);
cur->right=construct(inorder,i+1,iright,postorder,pleft+i-ileft,pright-1);
return cur;
}
};