题解 | #重建二叉树#
重建二叉树
http://www.nowcoder.com/practice/8a19cbe657394eeaac2f6ea9b0f6fcf6
算法分析
这道题目的描述,是需要我们从前序遍历和中序遍历推出整颗二叉树,以题目中的为实例开始分析。
前序遍历{1,2,4,7,3,5,6,8}
中序遍历{4,7,2,1,5,3,8,6}
前序的顺序为:根->左->右。 中序的顺序为:左->根->右。
我们结合图分析出,在中序遍历当中,每次树的根节点左边为左树元素,右边为右树元素。这样就可以用根节点进行划分左右树的元素,再把划分后的树递归再次划分,直至左右子树均为空
编写递归
public TreeNode reConstructBinaryTree(int [] pre,int [] vin) {
if(pre==null||pre.length==0)//判断传入数组是否为null,为空则当次递归中止。
return null;
TreeNode head=new TreeNode(pre[0]);
int index = find(pre[0],vin);//每次以前序遍历的元素作为根划分左右树。
head.left=reConstructBinaryTree(Arrays.copyOfRange(pre,1,index+1),//取出树的左边部分数组
Arrays.copyOfRange(vin,0,index));//取出中序根节点左边部分,及左边的树
head.right=reConstructBinaryTree(Arrays.copyOfRange(pre,index+1,pre.length),//取出树的左边部分数组
Arrays.copyOfRange(vin,index+1,vin.length));//取出中序根节点右边部分,及右边的树
return head;
}
public int find(int n,int[] nums){//这个是判断n在数组中的位置,应该也不用解释
for (int i=0;i< nums.length;i++){
if(n==nums[i])
return i;
}
return 0;
}