题目描述:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路:
1.前序遍历第一个元素一定为根节点
2.在中序遍历中根节点左边的节点为左子树,由该节点数可知左子树的前序序列
(如该前序第一个节点为1,则根节点为1,对应中序中1之前的4,7,2这3个元素为左子树,
则可以确定前序中第一个元素后的3个元素序列2,4,7为左子树的前序)
3.根节点右边的为右子树
4.由此找到左子树和右子树的前序序列及中序序列然后不断递归
实现:
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
if(pre==null||in==null)
return null;
TreeNode root=reConstructBinaryTree(pre,0,pre.length-1,in,0,in.length-1); //
return root;
}
private TreeNode reConstructBinaryTree(int [] pre,int startPre,int endPre,int [] in,int startIn,int endIn)
{
if(startPre>endPre||startIn>endIn) //前序序列或中序序列长度等于0时递归结束
return null;
TreeNode root=new TreeNode(pre[startPre]); //根节点为前序序列第一个
for(int i=startIn;i<=endIn;i++)
{
if(in[i]==pre[startPre]){
root.left=reConstructBinaryTree(pre,startPre+1,startPre+i-startIn,in,startIn,i-1);
//左子树的前序序列及中序序列
root.right=reConstructBinaryTree(pre,i-startIn+startPre+1,endPre,in,i+1,endIn);
//右子树的前序序列及中序序列
break;
}
}
return root;
}
}