题目
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。 例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
分析
前序遍历的顺序是“根->左->右”,中序遍历的顺序是“左->根->右”。因此,前序遍历的第一个数就是二叉树的根节点,在中序遍历中, 找到根节点,根节点的左边是二叉树的左子树,根节点的右边是二叉树的右子树。上面的问题中,根节点是1,在中序遍历中找到1, 那么{4,7,2}是二叉树的左子树,{5,3,8,6}是根节点的右子树。
我们将{4,7,2},{5,3,8,6}看做一颗新的二叉树,再用上面的思路进行根节点的寻找,就可以重构整个二叉树。
代码
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
import java.util.Arrays;
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
if (pre.length == 0 || in.length == 0 || pre == null || in == null) {
return null;
}
// 根节点
TreeNode root = new TreeNode(pre[0]);
// 寻找和根节点相同的中序遍历的点
for(int i =0;i<in.length;i++){
if (in[i]==root.val) {
// 在中序遍历找到根节点后,将根节点左边的和右边的序列分别看成新的子树,进行遍历,注意AArrays.copyOfRange(T[ ] original,int from,int to)
//将一个原始的数组original,从小标from开始复制,复制到小标to,生成一个新的数组。注意这里包括下标from,不包括下标to。
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;
}
}