重建二叉树
重建二叉树
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
分析
通过前序遍历和中序遍历去重建二叉树
前序遍历代表根在最前方,中序遍历根在其左子树和右子树的中间。
我们通过创建根节点的方式去重建整个二叉树,
要用根来创建二叉树,需要对前序进行遍历,整个问题可以化分为数个子问题进行解决。
创建一个二叉树,首先创建根节点,其次创建左子树和右子树。
从上面数据,首先找到根节点->1,其在中序遍历的位置4,7,2,->1->5,3,8,6.
其左子树为4,7,2.右子树为5,3,8,6.
此时,由先序遍历可知,1的左子树为2,4,7.由此可以得出,2为1的左子根,
而1的右子根从前序遍历中在左子树的节点之后,右子根为3.
继续划分子问题,将先序遍历中1的左右子树也找出来,
左子树2,4,7,右子树为3,5,6,8
先将左子树按照上述方式进行创建,
2划分左右子树,4,7为2的左子树,在中序遍历中2的右边为1,因此没有右子树。
在先序遍历中2之后为4,且在中序遍历中4在2的左边,因此4为2的左子树,
又有7为4的子树,7在4的右边,7为4的右子树。
以此推导,按照根在先序遍历中的顺序来创建节点,按照中序遍历中的顺序来判断其是左子树还是右子树。
代码如下:
import java.util.Arrays;
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
if(pre.length==0||in.length==0){
return null;
}
TreeNode root = new TreeNode(pre[0]);
buildTree(root,pre,in);
return root;
}
public void buildTree(TreeNode root,int[] pre,int[] in){
int i =0;
for(;i<in.length;i++){
if(in[i]==root.val){
break;
}
}
if(i>0){
root.left = new TreeNode(pre[1]);
buildTree(root.left,Arrays.copyOfRange(pre,1,i+1),Arrays.copyOfRange(in,0,i));
}
if(in.length-i-1>0){
root.right = new TreeNode(pre[i+1]);
buildTree(root.right,Arrays.copyOfRange(pre,i+1,pre.length),Arrays.copyOfRange(in,i+1,in.length));
}
}
}