输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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; }
* }
*/
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
if(pre==null || in ==null || in.length<0 ||pre.length<0){
return null;
}
return contruct(pre,0,pre.length-1,in,0,in.length-1);
}
public static TreeNode contruct(int [] pre,int ps,int pe,int [] in,int is,int ie){
// 开始位置大于结束位置说明已经没有需要处理的元素了
if(ps>pe){
return null;
}
int value=pre[ps];
int index=is;
while(is<ie && in[index]!=value){
index++;
}
if(index>ie){
throw new RuntimeException("Invalid input");
}
TreeNode node=new TreeNode(value);
// 递归构建当前根结点的左子树,左子树的元素个数:index-is+1个
// 左子树对应的前序遍历的位置在[ps+1, ps+index-is]
// 左子树对应的中序遍历的位置在[is, index-1]
node.left=contruct(pre,ps+1,ps+index-is,in,is,index-1);
// 递归构建当前根结点的右子树,右子树的元素个数:ie-index个
// 右子树对应的前序遍历的位置在[ps+index-is+1, pe]
// 右子树对应的中序遍历的位置在[index+1, ie]
node.right=contruct(pre,ps+index-is+1,pe,in,index+1,ie);
return node;
}
}