输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如:输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
- 在先序数组中,第一个数就是这棵二叉树的根节点
- 找到这个数在中序数组中的位置
- 这个位置之前的数都是左子树的,之后的都是右子树的
- 递归左子树,返回左子树根节点,父节点的左子节点指向该根节点
- 递归右子树,返回右子树根节点,父节点的右子节点指向该根节点
- 直到左右子树递归完成,停止递归
public class BuilderTree {
//节点的定义
public static class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
//主方法
public static TreeNode reConstructBinaryTree(int [] pre,int [] in) {
return builderTree(pre,in,0,pre.length - 1,0,in.length - 1);
}
//递归方法
private static TreeNode builderTree(int[] pre,int[] in,int preStart,int preEnd,int inStart,int inEnd) {
if (preStart >= preEnd) {
return null;
}
//根节点的值
int rootVal = pre[preStart];
//找到中序数组中根节点的位置
int find = inStart;
while(find <= inEnd){
if(rootVal == in[find]){
break;
}else{
find++;
}
}
//左子树的长度
int len = find - inStart;
//建立根节点
TreeNode root = new TreeNode(rootVal);
//递归左子树,返回左子树的根节点
root.left = builderTree(pre,in,preStart + 1,preStart + len,inStart,find - 1);
//递归右子树,返回右子树的根节点
root.right = builderTree(pre,in,preStart + 1 + len,preEnd,find + 1,inEnd);
//返回根节点
return root;
}
}