题目:重建二叉树
题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
解题思路
二叉树:二叉树是树的一种特殊结构,在二叉树中每个结点最多只能有两个子结点。在二叉树中最重要的操作是遍历,即按照某一顺序访问树中的所有结点。树的遍历方式:
前序遍历:先访问根节点,再访问左子结点,最后访问右子结点;(根左右)
中序遍历:先访问左子结点,再访问根结点,最后访问右子结点;(左根右)
后序遍历:先访问左子结点,再访问右子结点,最后访问根结点;(左右根)
在二叉树的前序遍历序列中,第一个数字总是树的根结点的值。但在中序遍历序列中,根结点的值在序列的中间,左子树的结点的值位于根结点的值的左边,而右子树的结点的值位于根结点的值的右边。因此我们需要扫描中序遍历序列,才能找到根结点的值。
前序遍历序列的第一个数字1就是根结点的值。扫描中序遍历序列,就能确定根结点的值的位置。根据中序遍历特点,在根结点的值1前面的3个数字都是左子树结点的值,位于1后面的数字都是右子树结点的值。
分别找到了左、右子树的前序遍历序列和中序遍历序列,我们就可以用同样的方法分别去构建左右子树,这是一个递归的过程。
参考代码
/*
public class TreeNode
{
public int val;
public TreeNode left;
public TreeNode right;
public TreeNode (int x)
{
val = x;
}
}*/
class Solution
{
public TreeNode reConstructBinaryTree(int[] pre, int[] tin)
{
// write code here
if(pre == null || tin == null)
return null;
return reConstructBinaryTree(pre,0,pre.Length-1,
tin,0,tin.Length-1);
}
private TreeNode reConstructBinaryTree(
int[] pre,int startPre,int endPre,int[] tin,int startIn,int endIn) {
if(startPre>endPre||startIn>endIn) //停止递归的条件
return null;
//前序遍历数组的第一个元素就是子树的根节点
TreeNode tree = new TreeNode(pre[startPre]);
for(int i=startIn;i<=endIn;i++)
{
if(tin[i]==pre[startPre])
{
// 其中(i - startIn)为中序排序中左子树结点的个数
//左子树
tree.left=reConstructBinaryTree(
pre,startPre+1,startPre+i-startIn,tin,startIn,i-1);
//右子树
tree.right=reConstructBinaryTree(
pre,i-startIn+startPre+1,endPre,tin,i+1,endIn);
}
}
return tree;
}
}