题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路:
前序:根左右,中序:左根右
题中例子:前序遍历1为根,中序遍历的4,7,2即为左子树,5,3,8,6即为右子树
左子树前序遍历为2,4,7,右子树前序遍历为3,5,6,8。左子树的中序遍历为4,7,2,右子树的中序遍历为5,3,8,6
对于左子树和右子树又可以采用以上思路不断划分。因此可以使用递归
public class Solution1 {//递归调用,重建二叉树
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
return helpReConstruct(pre, 0, pre.length-1, in, 0, in.length-1);
}
private TreeNode helpReConstruct(int[] pre,int startPre,int endPre,int[] in,int startIn,int endIn) {
if(startPre>endPre || startIn>endIn) {
return null;
}
TreeNode node = new TreeNode(pre[startPre]);
for(int i = startIn;i<=endIn;i++) {
if(node.val == in[i]) {
node.left = helpReConstruct(pre, startPre+1, startPre+i-startIn, in, startIn, i-1);
node.right = helpReConstruct(pre, startPre+i-startIn+1, endPre, in, i+1, endIn);
}
}
return node;
}
}