一、题目描述:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
二、思路解析
考察已知前序和中序来构造二叉树。二叉树是特殊的,本身也是递归定义的,所以关于二叉树的很多考题都是从递归来考虑的,仔细分析此题,不难发现,前序遍历的第一个结点就是该二叉树的根节点,再将该节点带入中序遍历中,就可以分出左子树和右子树了,然后在将左子树带入前序遍历中,第一个结点就是该左子树的根节点,然后在将右子树带入前序遍历中,第一个结点就是该右子树的根节点,以此类推,归纳出递归求解。
三、java代码(可运行)
package 剑指offer;
import javax.xml.soap.Node;
public class Solution_4 {
/*
* 二叉树节点类
*/
class TreeNode{
int val;
TreeNode left;
TreeNode right;
TreeNode(int x){
this.val = x;
}
}
//构建二叉树
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
if(pre == null || in == null || pre.length != in.length){
return null;
}
return construct(pre, 0, pre.length-1, in, 0, in.length-1);
}
/**
* @param pre 前序遍历
* @param ps 前序遍历的开始位置
* @param pe 前序遍历的结束位置
* @param in 中序遍历
* @param is 中序遍历的开始位置
* @param ie 中序遍历的结束位置
* @return 数的根节点
*/
//递归构建二叉树
private TreeNode construct(int [] pre,int ps,int pe,int [] in,int is,int ie){
if(ps>pe) return null;
int value=pre[ps]; //取前序遍历的第一个数字就是根节点
int index = is; // 在中序遍历中中寻找根节点
while(index<=ie && in[index]!=value){
index++;
}
if(index>ie){
System.out.println("Invalid Input");
return null;
}
TreeNode node = new TreeNode(value);
node.left = construct(pre, ps+1, ps+index-is, in, is, index-1);
node.right= construct(pre, ps+index-is+1, pe, in, index+1, ie);
return node;
}
//中序遍历
private void printTree(TreeNode root){
if(root!=null){
printTree(root.left);
System.out.println(root.val+" ");
printTree(root.right);
}
}
public static void main(String[] args) {
Solution_4 tes = new Solution_4();
int pre[] = {1,2,4,7,3,5,6,8}; //二叉树的前序遍历
int in[] = {4,7,2,1,5,3,8,6}; //二叉树的中续遍历
TreeNode node = tes.reConstructBinaryTree(pre, in); //重建二叉树
tes.printTree(node);
}
}