1.题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
2.思路
根据前序遍历的节点位置情况进而推算,二叉树的结构特别是作为父节点的结构。从而对中序遍历的数组进行有序的划分。最后进行递归求解。
3.代码如下
import java.util.*;
public class Solution {
private HashMap<Integer,Integer> map=new HashMap<>();
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
for(int i=0;i<in.length;i++)
{
map.put(in[i],i);
}
return reconstructTree(pre,0,pre.length-1,0);
}
public TreeNode reconstructTree(int[] pre,int prel,int prer,int inl){
if(prel>prer)
{
return null;
}
TreeNode node=new TreeNode(pre[prel]);
int indexin=map.get(node.val);
int leftTreelen=indexin-inl;
node.left=reconstructTree(pre,prel+1,prel+leftTreelen,inl);
node.right=reconstructTree(pre,prel+leftTreelen+1,prer,inl+leftTreelen+1);
return node;
}
}
注:通过HashMap结构进行对中序遍历节点的存储从而方便对已知节点的值,从而求出他在中序遍历数组中的位置,进而求解。