输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
题目分析:
两个遍历的规则如下
前序遍历 根,左,右,
中序遍历, 左,根 ,右
前序遍历:ABCDEFGHK
中序遍历:BDCAEHGKF,从前序遍历中,我们可以先获取到根节点的值。由于值不重复,我们可以从中序遍历中划分开左右子树的数值,然后通过递归的方法重新构建二叉树,将每次获取到的新节点都当成一个根来进行处理
import java.util.*;
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
//前序遍历 根,左,右,
//中序遍历, 左,根 ,右
if(pre.length == 0 || in.length == 0 ){
return null;
}
int root = pre[0];
int length = in.length;
TreeNode node = new TreeNode(root);
//获取根在in 的下标
for(int i = 0 ; i < length;i++){
if(in[i] == root){
//pre 1~i+1 为下一个根起点,到in 中的root位置 在对应的 0~i 中
//左子树的根 为pre[1],左子树的长度从中序遍历中取得。0-根节点的位置,左子树的中序遍历为in的 0到根节点
//右子树相反
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,length));
}
}
return node;
}
}