试题:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
代码:
树之二叉树遍历先序,中序,后序
根据二叉树先序和中序的遍历规则,可以知道我们可以首先根据先序确定根节点,然后根据中序判断出当前根节点的左树节点和右树节点。这时候问题又变成对左树节点和右树节点重建二叉树。很显然问题变成了递归问题,首先写出递归代码。然后确定如何根据递归顺序对各节点访问数据生成二叉树。由于二叉树有左右节点关系之分,最好办法就是在调用递归函数时返回节点信息从而确定左右关系。
import java.util.*;
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
//递归到底返回条件
if(pre.length==0 || in.length==0){
return null;
}
TreeNode node = new TreeNode(pre[0]);
for(int i=0; i<in.length; i++){ //查找根节点分界处
if(pre[0] == in[i]){
node.left = reConstructBinaryTree(Arrays.copyOfRange(pre, 1, i+1), Arrays.copyOfRange(in, 0, i)); //加1考虑到先序根节点位置
node.right = reConstructBinaryTree( Arrays.copyOfRange(pre, i+1, pre.length), Arrays.copyOfRange(in, i+1, in.length) ); //加1考虑到中序根节点位置
}
}
// 递归节点极其子树返回
return node;
}
}