问题描述: 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路: 从先序遍历中取出第一个元素即树的根结点,然后在中序遍历中以根结点为基准划分出根节点的左右子树的中序遍历序列,通过计算中序遍历中根结点左右子树结点数量可以得出左右子树的先序遍历序列,因此可以得到左子树和右子树的前序遍历和中序遍历的结果。符合构成递归需具备的条件:
- 子问题须与原始问题为同样的事,且更为简单;
- 不能无限制地调用本身,须有个出口,化简为非递归状况处理。
代码:
import java.util.Arrays;
import org.junit.Test;
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) {
val = x;
}
}
public class ReConstructBinaryTree {
//solution
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
if(pre.length==0)
return null;
TreeNode root = new TreeNode(pre[0]);
for(int i = 0 ; i < in.length;i++){
if(in[i]==pre[0]){
//copyOfRange复制Arrays的[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));
break;
}
}
return root;
}
//先序遍历
public void printPre(TreeNode root){
if(root!=null)
System.out.print(root.val+" ");
if(root.left!=null)
printPre(root.left);
if(root.right!=null)
printPre(root.right);
}
//中序遍历
public void printIn(TreeNode root){
if(root.left != null)
printIn(root.left);
if(root!=null)
System.out.print(root.val+" ");
if(root.right != null)
printIn(root.right);
}
@Test
public void test(){
int[]pre = {1,2,4,7,3,5,6,8};
int[]in = {4,7,2,1,5,3,8,6};
TreeNode root = reConstructBinaryTree(pre,in);
printPre(root);
System.out.println();
printIn(root);
}
输出:
1 2 4 7 3 5 6 8
4 7 2 1 5 3 8 6