剑指offer----重建二叉树---java实现

题目:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

思路:根据先序遍历数组,可以得到树的根节点,根据得到的根节点,去中序数组中找到相应的根节点,因为中序的遍历顺序是左----右,可以得到根节点的左子树和右子树的中序遍历数组,这样树的左子树中节点的个数和右子树上节点的个数就确定了。在先序数组中,左子树的先序和左子树的先序也可以确定了。

参数:先序遍历数组,子树在先序数组中的开始位置,结束位置。

         中序遍历数组,子树在中序数组中的开始位置,结束位置。

可以根据先序和中序数组得到子树的先序和中序数组,所以此问题可以用递归解决。

代码:

/*class TreeNode 
 {
	     int val;
	     TreeNode left;
         TreeNode right;
	     TreeNode(int x) { val = x; }
 }
*/
public class BulidTree
{
	
	 public TreeNode reConstructBinaryTree(int [] pre,int [] in)
	 {
		 if(pre == null || in == null || pre.length != in.length )//如果先序或者中序数组有一个为空的话,就无法建树,返回为空
			 return null;
		 else
		 {
			 return reBulidTree(pre,0,pre.length-1,in,0,in.length-1);
		 }
	 }
	 private TreeNode reBulidTree(int[] pre,int startPre,int endPre,int[] in,int startIn,int endIn)
	 {
		 if(startPre > endPre || startIn > endIn)//先对传的参数进行检查判断
			 return null;
		 int root = pre[startPre];//数组的开始位置的元素是跟元素
		 int locateRoot = locate(root,in,startIn,endIn);//得到根节点在中序数组中的位置 左子树的中序和右子树的中序以根节点位置为界
		 if(locateRoot == -1) //在中序数组中没有找到跟节点,则返回空
		   return null;
		 TreeNode treeRoot = new TreeNode(root);//创建树根节点
		 treeRoot.left = reBulidTree(pre,startPre + 1,startPre + locateRoot - startIn,in,startIn,locateRoot-1);//递归构建左子树
		 treeRoot.right = reBulidTree(pre,startPre+locateRoot-startIn+1,endPre,in,locateRoot+1,endIn);//递归构建右子树
		 return treeRoot;
	 }
	 //找到根节点在中序数组中的位置,根节点之前的是左子树的中序数组,根节点之后的是右子树的中序数组
	 private int locate(int root,int[] in,int startIn,int endIn)
	 {
		 for (int i = startIn; i < endIn; i++)
		{
			if(root == in[i])
				return i;
		}
		 return -1;
	 }
}


  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值