题目描述:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路:前序遍历,第一个数为根节点,确定好根节点后,在中序遍历里面找到根节点。位于根节点左边的就是该树的左子数,位于根节点右边就是右子数。分别算出左子树和右字数的个数。在前序遍历中,找出以找出节点为根节点的前序序列,在中序遍历中,找到以此节点为根节点的中序序列。右子树也类似。然后在找到子序列的根节点,递归调用,直到为空。
这里注意两个判断条件。
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in){
if(pre==null || in==null ){
return null;
}
TreeNode root = new TreeNode(pre[0]);//根节点
int num = 0;
for(int i = 0;i<in.length;i++){
if(in[i]!=root.val){
num++;
}else{
break;
}
}
int rightlenth = in.length-1-num; //右子数的个数
int[] subpre = new int[num]; //子前序
int []subin = new int[num]; //子中序
if(num>0){
for(int i = 0;i<num;i++){
subpre[i] = pre[i+1];
subin[i] = in[i];
}
root.left = reConstructBinaryTree(subpre,subin);
// root.setLeft(left);
}
int rightpre[] = new int[rightlenth];
int rightin[] = new int[rightlenth];
if(rightlenth>0){
for(int i =0;i<rightlenth;i++){
rightpre[i] = pre[num+i+1];
rightin[i] = in[num+1+i];
}
root.right = reConstructBinaryTree(rightpre,rightin);
}
//root.left =reConstructBinaryTree(subpre,subin);
return root;
}
}
只有当左子树或者右子数个数大于0的时候,才会找出前序序列和中序序列。
2019/6/13 打卡