重建二叉树
题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
前序遍历:是先根节点,然后是左子树然后是右子树,然后左子树和右子树都循环这个规则
中序遍历:按照左中右的顺序进行
后序遍历:按照左右中的顺序进行
其中用到Arrays类,该类包含用来操作数组的(排序,搜索)等各种方法,其中Arrays.copyofRange(数组,int from,int to)方法是将指定数组的指定范围复制到一个新数组里面。范围是左闭右开(左边包含,右边不包含),这个很重要!!
解题思路:先找到中序中和前序第一个结点相同的节点即为分界点,中序中该点左边的部分为左子树,右边为右子树,然后依次递归该方法,知道找到所有的节点。
解题代码:
public class BinaryTree {
public static class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
public static TreeNode reConstructBinaryTree(int [] pre,int [] in) {
TreeNode treeNode = null;
for(int i = 0;i < in.length; i++){
if(in[i]==pre[0]){
treeNode = new TreeNode(in[i]);
treeNode.left = reConstructBinaryTree(Arrays.copyOfRange(pre,1,i+1),Arrays.copyOfRange(in,0,i));
treeNode.right = reConstructBinaryTree(Arrays.copyOfRange(pre,i+1,pre.length),Arrays.copyOfRange(in,i+1,in.length));
}
}
return treeNode;
}
public static void main(String args[]){
int[] pre = {1,2,3,4,5,6,7};
int[] in = {3,2,4,1,6,5,7};
}
}