剑指offer第4题:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
理解:
本题首先应当理解前序遍历和中序遍历是什么,这里的“前”和“中”都是指遍历根节点的次序,前序遍历是先遍历根节点,中序遍历是先遍历左子节点然后才是根节点。
给出前序遍历和中序遍历,或者给出后序遍历和中序遍历,都是可以还原树的,但只给前序和后序遍历是不可以还原的,因为不能找到根节点的位置。
本题利用递归的思想,首先把前序遍历的第一个元素的值赋给根节点;然后找到中序遍历中根节点的位置,根据这个位置一切为二,在左子树和右子树分别递归调用。用到了copyOfRange(arrays, start, end)方法,可以将数组中的一段拿出来,在这里使用很方便。需要注意的是后面end是开区间。代码如下:
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
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 || pre.length != in.length){ //如果长度非法返回空
return null;
}
TreeNode newTree = new TreeNode(pre[0]); //前序遍历的第一个为根节点
for(int i = 0; i < in.length; i++){
if(pre[0] == in[i]){ //找到中序遍历中的根节点的位置
//递归操作,将左子树和右子树分别递归执行重建树的程序
//用到copyOfRange(操作对象,开始坐标,结束坐标),注意结束坐标是开区间,开始是闭区间
newTree.left = reConstructBinaryTree(Arrays.copyOfRange(pre,1,i+1),Arrays.copyOfRange(in,0,i));
newTree.right = reConstructBinaryTree(Arrays.copyOfRange(pre,i+1,pre.length),Arrays.copyOfRange(in,i+1,in.length));
}
}
return newTree;
}
}
知识点:
- TreeNode.left; TreeNode.right;
- Arrays.copyOfRange(arrays, start, end);
- Arrays需要import java.util.Arrays;来使用。