剑指offer-重建二叉树

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

解题思路:
在二叉树的前序遍历中,第一个数字总是树的根节点,但在中序遍历中,根节点的值在序列的中间,左子树的序列在根节点的左侧,右子树的序列在根节点的右侧。因此需要在中序遍历中找到根节点的位置,以确定前序遍历和中序遍历中左子树和右子树的序列。树的构建可使用递归构造。
tips:在中序序列中定位树的根节点的位置,使用HashMap可使得其过程复杂度为O(1),<中序序列数值,中序序列数值所对应的下标>。

 public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        //特殊情况
		Map<Integer, Integer> map = new HashMap<>();
		for(int i=0;i<in.length;i++){
			map.put(in[i], i);
		}
		int len_pre = pre.length;
		int len_in = pre.length;
		if(pre==null||in==null||len_in<=0||len_pre<=0||len_in!=len_pre){
			return null;
		}//特殊边界判定
		return createTree(pre, 0, len_pre-1, in, 0, len_in-1,map);
    }

	private TreeNode createTree(int[] pre, int start_pre, int end_pre, int[] in, int start_in, int end_in,Map<Integer, Integer> map) {
		// TODO Auto-generated method stub
		//前序的第一个元素就是根节点,创建根节点
		//递归停止条件
		if(start_pre>end_pre){
			return null;
		}
		TreeNode root = new TreeNode(pre[start_pre]);

		//查找根节点在中序中的位置,使用HashMap降低时间复杂度
		int rootIndex = map.get(pre[start_pre]);
		int leftLen = rootIndex - start_in;
		int pre_left = start_pre + leftLen;
		root.left = createTree(pre, start_pre+1, pre_left, in, start_in, rootIndex-1, map);
		root.right = createTree(pre, pre_left+1, end_pre, in, rootIndex+1, end_in, map);	
		return root;
	}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值