题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
/**
* Definition for binary tree 给出的二叉树的数据结构
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
时间限制:1秒 空间限制:32768K
PS:刚接触算法,小白一枚,对于有些题目是苦苦思索而不得。有时候有思路却无法写出代码,有时候连别人给出的答案也要想半天 :( 。遂将自己的理解记录于此,也给他人提供些解题的参考。
言归正传,要解决本题目,首先起码知道前序遍历,中序遍历吧,下面给出《数据结构》严蔚敏 吴伟民 书中的定义
前序遍历:若二叉树为空,则空操作;否则
- 访问根节点;
- 先序遍历左字树;
- 先序遍历右子树;
中序遍历:若二叉树为空,则空操作;否则
- 中序遍历左子树;
- 访问根节点;
- 中序遍历右子树;
OK,明白了这两个遍历,我们可以直接得出的结论是
- 二叉树的根节点就是前序遍历中的第一个节点,即 1 是根节点
- 中序遍历序列中的 4、7、2 是二叉树的左子树,5、3、8、6是二叉树的右子树
图 1
图 2
顺便给个结果图做比对:
图 3
那么,我们是否能够利用递归思想,每次都将左右两边的子树当作一根新的二叉树处理
- 找出二叉树的根节点
- 构建左子树----->找出该左子树的根节点,构建该左字树的左子树,构建该左子树的右子树.....直到序列中没有左子树的序列
- 构建右子树----->找出该右子树的根节点,构建该右字树的左子树,构建该右子树的右子树.....直到序列中没有右子树的序列
下面给出 欲风 同学的代码,我做了点注释
package com;
public class Solution {
public class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) {
val = x;
}
}
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
TreeNode root=reConstructBinaryTree(pre,0,pre.length-1,in,0,in.length-1);
return root;
}
/**
*
* @param pre 前序遍历序列的数组
* @param startPre 前序遍历序列中 新子树开始的序号的下标
* @param endPre 前序遍历序列中 新子树结束的序号的下标
* @param in 中序遍历序列的数组
* @param startIn 中序遍历序列中 新子树开始的序号的下标
* @param endIn 中序遍历序列中 新子树结束的序号的下标
* @return
*/
private TreeNode reConstructBinaryTree(int [] pre,int startPre,int endPre,int [] in,int startIn,int endIn) {
/**
* 作为退出递归的条件
* 当该子树在序列中的开始的下标大于结束的下标时,说明已经没有子树,将该子树置为null
*/
if(startPre>endPre||startIn>endIn)
return null;
/**
* 新子树的根节点(一定是当前前序遍历序列中的pre[startPre])
*/
TreeNode root=new TreeNode(pre[startPre]);
for(int i=startIn;i<=endIn;i++)
/**
根据前序遍历的序列和中序遍历的序列,找出新子树的根节点,并递归计算新子树下的左子树和右子数
i即是中序遍历序列中本树的根节点的位置
*/
if(in[i]==pre[startPre]){
/**
*startPre = startPre + 1 :startPre是前序遍历序列中本树的根节点,所以下次递归要从本树的子树序列开始
* endPre = startPre + i -startIn :下次递归本树的子树序列结束的位置,在中序遍历序列中i的左边的所有序列都是本树的左子树,所以i也是本树的左子树的个数(如图一)
*/
root.left=reConstructBinaryTree(pre,startPre+1,startPre+i-startIn,in,startIn,i-1);
/**
* 里面的加加减减好难解释啊...有种只可意会不可言传的感觉。。。
* 相信你们将程序单步调试几遍后就会明白
*/
root.right=reConstructBinaryTree(pre,i-startIn+startPre+1,endPre,in,i+1,endIn);
break;
}
return root;
}
}
给出一张递归图解,写了一半左右,举一反三吧 :)