根据二叉树遍历结果重构二叉树(前序遍历+中序遍历 / 中序遍历+后序遍历 / LeetCode真题)
Posted 月下赶路人
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了根据二叉树遍历结果重构二叉树(前序遍历+中序遍历 / 中序遍历+后序遍历 / LeetCode真题)相关的知识,希望对你有一定的参考价值。
文章目录
重构二叉树
所谓重构二叉树就是根据遍历结果构造还原出原本的二叉树,在二叉树的四种遍历方法(前序遍历、中序遍历、后序遍历、层序遍历)中已经对二叉树的遍历进行了详细的分析,接下来我们就要分析如何根据不同的遍历结果重构出二叉树。
前序遍历的访问顺序:根节点——>前序遍历左子树——>前序遍历右子树
中序遍历的访问顺序:中序遍历左子树——>根节点——>中序遍历右子树
后序遍历的访问顺序:后序遍历左子树——>后序遍历右子树——>根节点
根据三种遍历的访问顺序,我们可以得到如下图关系:
前序遍历+中序遍历
由上图可知,我们每一次都可以根据前序遍历得到二叉树的根,并在中序遍历中通过根确定左子树和右子树,循环往复,就可以确定每一个元素在二叉树中的位置。
我们以LeetCode105题为例: 从前序与中序遍历序列构造二叉树
根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/ \\
9 20
/ \\
15 7
代码实现:
public TreeNode buildTree(int[] preorder, int[] inorder)
int preStart = 0;//假指针,指向preOrder
int inStart = 0;//指向inorder
int preEnd = preorder.length;//因为数组本身的操作比较麻烦,因此我们借用数组的长度来辅助作为终止指针
int inEnd = inorder.length;
return build(preorder, preStart,preEnd, inorder,inStart, inEnd);
private TreeNode build(int[] preorder,int preStart,int preEnd, int[] inorder,int inStart,int inEnd)
if (preStart == preEnd) //前序遍历数组没有数据了就返回
return null;
TreeNode root = new TreeNode(preorder[preStart]);
//从中序遍历中寻找根节点的位置
int index = inStart;
for (int i = inStart; i < inEnd; i++)
if (preorder[preStart] == inorder[i])
index = i;
break;
//左子树数量
int leftNum = index - inStart;
//为根节点添加左右子树
root.left = build(preorder, preStart + 1, preStart+leftNum+1,
inorder, inStart, index);
root.right = build(preorder, preStart + leftNum + 1, preEnd,
inorder, index + 1, inEnd);
return root;
后序遍历+中序遍历
和前序遍历相同,我们每一次都可以根据后序遍历得到二叉树的根,并在中序遍历中通过根确定左子树和右子树,循环往复,就可以确定每一个元素在二叉树中的位置。
我们以LeetCode106题为例: 从中序与后序遍历序列构造二叉树
根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树:
3
/ \\
9 20
/ \\
15 7
代码实现:
这个实现方法和上面的实现方法基本上是一样的,但这里我们使用HashMap将中序遍历的结果存储起来方便查找
int postEnd;
HashMap<Integer, Integer> map = new HashMap<>();
public TreeNode buildTree(int[] inorder, int[] postorder)
for (int i = 0; i < inorder.length; i++) //将中序遍历结果存在map中,k为值,v为索引
map.put(inorder[i], i);//作用是方便获取根节点
int inStart = 0;
int inEnd = inorder.length-1;
this.postEnd= postorder.length-1;
return build(inorder, inStart, inEnd, postorder);
private TreeNode build(int[] inorder, int inStart, int inEnd,
int[] postorder)
if (inStart > inEnd)
return null;
//获取根节点
TreeNode root = new TreeNode(postorder[postEnd]);
//在中序遍历中拿到根节点所在的索引,区分左右子树
int index = map.get(postorder[postEnd]);
//postEnd减小
postEnd--;
//递归拿到左右子树
root.right = build(inorder, index+1, inEnd,postorder);
root.left = build(inorder, inStart,index-1,postorder);
return root;
前序遍历+后序遍历
前序和后序遍历结果重构二叉树是有限制的,即二叉树必须是一颗真二叉树(度为1或0,就是要么左右节点都有,要么都没有),否则重构的结果不唯一。原因是当我们的二叉树缺少左子树和右子树时,我们只能确定根节点,不能确定剩下来的元素属于左子树还是右子树。
以上是关于根据二叉树遍历结果重构二叉树(前序遍历+中序遍历 / 中序遍历+后序遍历 / LeetCode真题)的主要内容,如果未能解决你的问题,请参考以下文章
根据二叉树遍历结果重构二叉树(前序遍历+中序遍历 / 中序遍历+后序遍历 / LeetCode真题)
根据二叉树遍历结果重构二叉树(前序遍历+中序遍历 / 中序遍历+后序遍历 / LeetCode真题)