输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
/** * Definition for binary tree * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ import java.util.Arrays;//Arrays.copyOfRange(,,); //针对pre[]和in[]数组,功能:选择范围复制==>左闭右开区间 public class Solution { public TreeNode reConstructBinaryTree(int [] pre,int [] in) { //返回的是根节点 if(pre.length==0||in.length==0)return null;//【递归的终结】 //也可以简化为 pre.length==0 (只判断一个即可) //不能写成 pre==null,因为pre==[]时,数组不是null但长度为零 TreeNode node=new TreeNode(pre[0]);//先序的第一个pre[0]永远是根节点,也是分左右子树的关键 for(int i=0;i<pre.length;i++){ //pre和in的子数组永远是对应相同长度的 if(pre[0]==in[i]){//每一次for循环,只有一次会执行if里面的语句 node.left=reConstructBinaryTree(Arrays.copyOfRange(pre,1,i+1),Arrays.copyOfRange(in,0,i)); node.right=reConstructBinaryTree(Arrays.copyOfRange(pre,i+1,pre.length),Arrays.copyOfRange(in,i+1,in.length)); }//在建设node.val后,再递归调用获取node.left和node.right,这样3步后,一个node就完整建立起来了 } return node; } } //复杂度方面:最坏情况下(树是一条直线)每一层递归从O(n)直到O(1),因为每一层都会至少减少1个复制的。 //最坏情况下,N层递归,此时时间空间复杂度都是O(N^2) //平均情况下,log(N)层递归,此时时间空间复杂度都是O(NlogN)