题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路:了解二叉树的前,中,后序遍历的规律,我们就可以知道,通过前 中,或者 后 中 两种都能得到二叉树。主要通过三个步骤:
1.查看前(后)序遍历序列,最开始(后面)的必然是根节点。
2.获得根节点后,查看中序遍历序列,根节点左右两边分别为左子树的所有节点,和右子树的所有节点 4 7 2 ,5 3 8 6.
3.分别对两边序列重复 1 . 2 两个步骤。最终能得到二叉树。
代码:
import java.util.Arrays;
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
if (pre == null || in == null) {
return null;
}
if (pre.length == 0 || in.length == 0) {
return null;
}
if (pre.length != in.length) {
return null;
}
TreeNode root = new TreeNode(pre[0]);
for (int i = 0; i < pre.length; i++) {
if (pre[0] == in[i]) {
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));
}
}
return root;
}
}