重构二叉树
题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
1.递归法
根据前序遍历序列的第一个节点可以得到该二叉树的根节点,
根据根节点在中序遍历序列中的位置可以把二叉树分为左右两颗子树
再分别对左右两颗子树进行递归求解
例如:我现在有
前序遍历序列[1,2,4,7,3,5,6,8],
中序遍历序列[4,7,2,1,5,3,8,6],要想实现重构二叉树,我们可以先获取前序序列中的第一个元素1,然后在中序遍历序列中查找1所在的位置,可以看出它在数组下标为3的位置上,以1为界可以把二叉树分成左右两部分,左边的为左子树,右边的为又子树。此时,
左子树
前序遍历序列[2,4,7],
中序遍历序列[4,7,2],
右子树
前序遍历序列[3,5,6,8],
中序遍历序列[5,3,8,6],
在利用递归分别对左右子树进行重构,直到数组的长度为0
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]);
for(int i = 0; i< in.length; i++){
if(in[i] == pre[0]){
root.left = reConstructBinaryTree(Arrays.copyOfRange(pre,1,i+1),Arrays.copyOfRange(in,0,i));
root.right = reConstructBinaryTree(Arrays.copyOfRange(pre,i+1,pre.length),Arrays.copyOfRange(in,i+1,in.length));
break;
}
}
return root;
}
}