重建二叉树

19 篇文章 0 订阅
19 篇文章 0 订阅

剑指offer第六题,自己很快得出了由先序和中序遍历得出树的结构的思路,但是实现起来遇到了一些困难,花了不少时间

package com.zjy.sword2offer;

class TreeNode{
	int val;
	TreeNode left;
	TreeNode right;
}

public class ReconstructTree {
	public TreeNode ReconstrTree(int[] a, int[] b) throws Exception{
		if(a==null||b==null||a.length<=0||b.length<=0)
			return null;
		
		if(a.length!=b.length)
			return null;
		return ReconstrTreeCore(a, 0,a.length-1, b,0,a.length-1);
	}
	
	public TreeNode ReconstrTreeCore(int[] a,int prestart,int preend,
int[] b,int instart,int inend) throws Exception{
		int rootvalue = a[prestart];
		TreeNode root = new TreeNode();
		root.val = rootvalue;
		root.left = root.right =null;
		
		if(prestart==preend)
		{
			if(instart==inend && a[preend]==b[inend])
				return root;
			else
				throw new Exception("invalid input");
		}
		
		int rootinorder = instart;
		while(rootinorder<inend && a[rootinorder]!=rootvalue)
			rootinorder++;
		
		if(rootinorder==inend&&a[rootinorder]!=rootvalue)
			new Exception("invalid input");
		
		int leftTreeLen = rootinorder - instart;
		if(leftTreeLen>0)
		{
			root.left = ReconstrTreeCore(a,prestart+1,prestart+leftTreeLen,
b,instart,rootinorder-1);
		}
		if(leftTreeLen<preend-prestart)
		{
			root.right = ReconstrTreeCore(a,prestart+leftTreeLen+1,preend,
b,rootinorder+1,inend);
		}
		
		return root;
	}
	
}


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值