编写一些高效率的函数,只使用指向二叉树,序列化二叉树 剑指offer
输入二叉树的一阶遍历和中阶遍历的结果。请重建二叉树。
注意:
二叉树中每个节点的值互不相同;输入的前序遍历和中序遍历必须合法;样品
鉴于:
前序遍历是:[3,9,20,15,7]
中序遍历是:[9,3,15,20,7]
返回:[3,9,20,null,null,15,7,null,null,null]
返回的二叉树如下:
三
/\
9 20
/\
5 7个想法(递归)
递归构建整个二叉树:先递归创建左右两个子树,然后创建根节点,让指针指向这两个子树。
具体步骤如下:
通过前序遍历找到根节点:前序遍历的第一个数就是根节点的值;如果在中序遍历中找到根节点位置K,K的左边是左子树的中序遍历,右边是右子树的中序遍历;假设左子树的中序遍历长度为L,在前序遍历中,根节点后面的L个数为左子树的前序遍历,剩余个数为右子树的前序遍历;有了左右子树的前序遍历和中序遍历,就可以先递归创建左右子树,再创建根节点;时间复杂性分析
初始化时,我们用哈希表(unordered_map int,int)记录每个值在中序遍历中的位置,这样当我们递归到每个节点时,只需要花时间在中序遍历中找到根节点位置。此时,创建每个节点所需的时间为,因此总时间复杂度为。
代码/* *
*二叉树节点的定义。
*结构树节点{
* int val
* TreeNode * left
* TreeNode * right
* TreeNode(int x) : val(x),left(NULL),right(NULL) {}
* };
*/
类别解决方案{
公共:
unordered_map int,int pos
TreeNode* buildTree(矢量int preorder,矢量int inorder) {
int n=preorder . size();
for(int I=0;I n;我)
pos[in order[I]]=I;//记录根节点的位置
返回dfs(preorder,inorder,0,n - 1,0,n-1);//递归创建左右子树,最后将根节点指向左右子树。
}
TreeNode* dfs(向量整数预测,向量整数输入,整数预测,整数预测,整数预测,整数预测)
{
if (pl pr)返回NULL//如果左区间大于右区间,直接返回NULL
int k=pos[pre[pl]]-il;//前导遍历间隔的长度
TreeNode * root=new TreeNode(pre[pl]);//创建根节点
root- left=dfs(pre,in,pl 1,pl k,il,il k-1);//左子树的第一个和中间顺序遍历间隔
root- right=dfs(pre,in,pl k 1,pr,il k 1,IR);//右侧子树按照前中顺序遍历区间。
返回根目录;
}
};
郑重声明:本文由网友发布,不代表盛行IT的观点,版权归原作者所有,仅为传播更多信息之目的,如有侵权请联系,我们将第一时间修改或删除,多谢。