题目:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
解答:
①利用Arrays里的copyOfRange函数
注意copyOfRange函数,Arrays.copyOfRange(oringinal,int from, int to)包含from的元素,不包含to的元素。
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
//例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}
import java.util.Arrays;
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
if(pre==null||in==null){
return null;
}
if(pre.length==0||in.length==0){
return null;
}
if(pre.length!=in.length){
return null;
}
TreeNode root=new TreeNode(pre[0]);
for(int i=0;i<in.length;i++){
if(pre[0]==in[i]){
root.left=reConstructBinaryTree(Arrays.copyOfRange(pre,1,i+1),Arrays.copyOfRange(in,0,i));
//pre第0个元素就是这次的根,所以左子树为1开始,i是in的下标,是左子树节点的个数,i+1因为copyOfRange不包含后面的节点。in,0,i就是根节点左侧的左子树节点。
root.right=reConstructBinaryTree(Arrays.copyOfRange(pre,i+1,pre.length),Arrays.copyOfRange(in,i+1,in.length));
//pre,右子树i+1包含在里面,到pre.length-1,pre.length不包含。in里面,i的位置是根,都不包含,从n+1开始。
break;
}
}
return root;
}
}
运行时间: 261 ms
占用内存: 22744K
②拷贝数组会消耗大量内存,使用传入索引的方式
数字还在数组里面,只不过是通过索引的方式构造整个二叉树。
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
TreeNode root=reConstructBinaryTree(pre,0,pre.length-1,in,0,in.length-1);
return root;
}
//前序遍历{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}
private TreeNode reConstructBinaryTree(int [] pre,int startPre,int endPre,int [] in,int startIn,int endIn) {
if(startPre>endPre||startIn>endIn)
return null;
TreeNode root=new TreeNode(pre[startPre]);
for(int i=startIn;i<=endIn;i++)
if(in[i]==pre[startPre]){
root.left=reConstructBinaryTree(pre,startPre+1,startPre+i-startIn,in,startIn,i-1);
//pre和in一直没变过,startPre就是根节点,所以startPre+1开始是左子树的节点。i-startIn,i指的是中序遍历根节点的位置,减去中序遍历开始位置即左子树的长度。所以i-startIn+startPre是前序遍历左子树结束的位置。in就好理解了,startIn,i是根节点,中间就是左子树了。
root.right=reConstructBinaryTree(pre,i-startIn+startPre+1,endPre,in,i+1,endIn);
//i-startIn跟上面的一样,+startPre就是根节点的位置,再+1是右子树。In同样道理
break;
}
return root;
}
}
运行时间: 210 ms
占用内存: 21912K