题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
输入:[1,2,3,4,5,6,7],[3,2,4,1,6,5,7]
输出:{1,2,5,3,4,6,7}
1
/ \
2 5
/ \ / \
3 4 6 7
前置知识:如果右重复数字,则前序和中序无法确定一棵二叉树,因为不知道前序序列中的重复数字对应中序序列中同一重复数字的哪一个,导致无法判断当前前序序列中节点的左右节点。
前序(根->左->右),中序(左->根->右)
思路:
前序遍历的第一个节点为根节点,使用这个节点将中序遍历结果分成两部分,左部分为树的左子树中序遍历结果,右部分为树的右子树中遍历的结果。然后分别对左右子树递归求解。
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
//{4,7,2,1,5,3,8,6}中序遍历
//{1,2,4,7,3,5,6,8}前序遍历
/**
* 前序 中左右
* 中序 左中右
* 后续 左右中
*/
if(pre == null || pre.length == 0){//pre.length==0的话,in.length肯定也为0,判断一个就够了
return null;
}
return getRootNode(pre,0,pre.length-1,in,0,in.length-1);
}
private TreeNode getRootNode(int[] pre,int preFrom ,int preEnd,int[] in,int inFrom,int inEnd){
if(preEnd < preFrom || inFrom > inEnd){
return null;
}
TreeNode root = new TreeNode(pre[preFrom]);
int inIndex = inFrom;
for(int i=inFrom;i<=inEnd;i++){
if(in[i] == pre[preFrom]){
inIndex = i;
root.left = getRootNode(pre,preFrom+1,preFrom+inIndex-inFrom,in,inFrom,inIndex-1);
root.right = getRootNode(pre,preFrom+1+inIndex-inFrom,preEnd,in,inIndex+1,inEnd);
}
}
return root;
}
}