【剑指offer】重建二叉树
题目:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路:
在二叉树的前序遍历序列中,第一个数字总是树的根结点的值。但在中序遍历序列中,根结点的值在序列的中间,左子树的结点的值位于根结点的值的左边,而右子树的结点的值位于根结点的值的右边。因此我们需要扫描中序遍历序列,才能找到根结点的值。
如下图所示,前序遍历序列的第一个数字1就是根结点的值。扫描中序遍历序列,就能确定根结点的值的位置。根据中序遍历特点,在根结点的值1前面的3个数字都是左子树结点的值,位于1后面的数字都是右子树结点的值。
同样,在前序遍历的序列中,根结点后面的3个数字就是3个左子树结点的值,再后面的所有数字都是右子树结点的值。这样我们就在前序遍历和中序遍历两个序列中,分别找到了左右子树对应的子序列。
既然我们已经分别找到了左、右子树的前序遍历序列和中序遍历序列,我们可以用同样的方法分别去构建左右子树。也就是说,接下来的事情可以用递归的方法去完成。
代码:
Offer_4.java
package offer;
public class Offer_4 {
public TreeNode reConstructBinaryTree(int [] pre,int [] in){
TreeNode root = new TreeNode(pre[0]);//前序的第一个数定为根
int len = pre.length;
//当只有一个数的时候
if(len == 1){
root.left = null;
root.right = null;
return root;
}
//找到中序中 根的位置
int rootval = root.val;
int i;
for (i = 0;i < len; i++) {
if(rootval == in[i]){
break;
}
}
//创建左子树
if(i > 0){
int [] pr = new int[i];
int [] ino = new int[i];
for(int j = 0;j < i;j++){
pr[j] = pre[j+1];
}
for(int j = 0;j < i;j++){
ino[j] = in[j];
}
root.left = reConstructBinaryTree(pr, ino);
}else{
root.left = null;
}
//创建右子树
if(len - i - 1 > 0){
int[] pr = new int[len - i -1];
int[] ino = new int[len - i -1];
for(int j = i+1;j < len;j++){
ino[j-i-1] = in[j];
pr[j-i-1] = pre[j];
}
root.right = reConstructBinaryTree(pr, ino);
}else{
root.right = null;
}
return root;
}
}
TreeNode.java
package offer;
public class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x){val = x;}
}