输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回
- 分析
熟悉二叉树前序&中序遍历规则
前序遍历规则:遍历根节点—->遍历根节点的左子树—->遍历根节点的右子树
中序遍历规则:遍历根节点左子树—->遍历根的节点—->遍历根节点的右子树
前序:1 2 4 3 5 6 中序:4 2 1 5 3 6
从遍历规则可知:前序遍历结果:根节点一定是遍历的第一个结点
中序遍历结果:根节点一定在左子树之后右子树之前遍历
因此,可以从前序遍历结果中找到二叉树的根,从中序遍历结果中找到二叉树的左右子树
- 思路 :
a. 取前序遍历一个元素,即根节点中数据
b. 在中序遍历结果中找到a中所取元素的位置,按照该位置将序列分为左右两部分,即根的左右子树
c. 创建二叉树的根节点(数据为a中所取数据)
按照b中所划分区间:递归创建根左子树
/**
* Def inition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode *_reConstructBinaryTree(const vector<int>&pre,size_t &preIndex,const vector<int>&vin,size_t left,size_t right)
{
if(preIndex>=pre.size() || left>=right)
return NULL;
size_t rootIndex=left;
// 在中序遍历结果中找二叉树根的位置
while(rootIndex<right)
{
if(pre[preIndex]==vin[rootIndex])
break;
rootIndex++;
}
// 还原二叉树根节点
TreeNode *pRoot=new TreeNode(pre[preIndex]);
// 如果左区间存在,还原根的左子树
if(left<rootIndex)
pRoot->left=_reConstructBinaryTree(pre,++preIndex,vin,left,rootIndex);
// 如果右区间存在,还原根的右子树
if(rootIndex+1<right)
pRoot->right=_reConstructBinaryTree(pre,++preIndex,vin,++rootIndex,right);
return pRoot;
}
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
if(pre.size()!=vin.size())
return NULL;
size_t preIndex=0;
return _reConstructBinaryTree(pre,preIndex,vin,0,vin.size());
}
};