题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
样例分析
例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回{1,2,5,3,4,6,7}
思路分析
前序遍历序列中,根节点总在最前。而中序遍历序列中,根节点分割了左右子树,所以,找到中序遍历中的根节点,就知道了左右子树,依次思路进行递归,即可得出答案。
代码及结果
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
TreeNode root = reConstructBinaryTree(pre, 0, pre.length-1, in, 0, in.length-1);
return root;
}
public TreeNode reConstructBinaryTree(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 (in[i] == pre[startpre]) {
node.left = reConstructBinaryTree(pre, startpre+1, startpre+i-startin, in, startin, i-1);//不理解这里的,可借助长度理解,两数组中,左子树部分长度相同
node.right = reConstructBinaryTree(pre, startpre+i-startin+1, endpre, in, i+1, endin);
}
}
return node;
}