题目要求:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
解题思路:
1.先序遍历的第一个数值是根节点,从中序遍历中找到这个数字的位置,中序的位置左边是左子树上的所有节点,中序的位置右边是右子树的所有节点。
2.对左子树和右子树递归调用上述过程。(构建根节点,构建左右子树)
难点:
构建左右子树时,需要确定先序和中序的构建左右子树的起始范围
代码示例:
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
TreeNode root=reConstructBinaryTree(pre,0,pre.length-1,in,0,in.length-1);
return root;
}
//构建根节点的函数
public TreeNode reConstructBinaryTree(int[] pre,int startPre,int endPre,int[] in,int startIn,int endIn) {
if(startPre>endPre||startIn>endIn) {
return null;
}
//1.先序的第一个位置数值构建根节点 2.去中序数组中找到这个根节点所在的位置 3.划分先序遍历和中序遍历数组中左右子树的数组元素
//先序遍历中左子树的范围是:根节点的下一个(StartPre+1,startPre-startIn+i);中序遍历中左子树的范围是:startIn到i-1位置,元素一共是i-1-startIn个,所以先序遍历中左子树的范围是startPre+1+i-1-startIn,startPre-startIn+i
//先序遍历中右子树的范围是(endPre-endIn+1+i,endPre);中序遍历中右子树的范围是(i+1,endIn),共endIn-1-i个元素,所以开始位置是endPre-endIn+1+i,或者是左子树结束位置的下一个 i-startIn+startPre+1;
TreeNode root=new TreeNode(pre[startPre]);
for (int i = 0; i < in.length; i++) {
if (pre[startPre]==in[i]) {
root.left=reConstructBinaryTree(pre, startPre+1, startPre+i-startIn, in, startIn, i-1);
root.right=reConstructBinaryTree(pre, i-startIn+startPre+1, endPre, in, i+1, endIn);
}
}
return root;
}