题目描述:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。
假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如输入前序遍历序列{1,2,4,7,3,5,6,8}
和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回
代码实现:
/**
* @param pre
* 二叉树的前序遍历结果
* @param in
* 二叉树的中序遍历结果
* @return root 重构完成的二叉树的根节点
*/
public TreeNode reConstructBinaryTree(int[] pre, int[] in) {
// 前序遍历或者中序遍历的结果已构建完, 递归退出
if (pre == null || in == null || pre.length == 0 || in.length == 0) {
return null;
}
// 用前序遍历的结果的第一个元素, 构造二叉树的根节点
TreeNode root = new TreeNode(pre[0]);
int i;
// 在中序遍历的结果中,找出根节点pre[0]所在下标
for (i = 0; i < in.length; i++) {
if (in[i] == pre[0]) {
break;
}
}
// 构建当前根节点的左子树时, 前序遍历的数组为pre[1..i]
int[] leftPre = Arrays.copyOfRange(pre, 1, i + 1);
// 构建当前根节点的左子树时, 中序遍历的数组为in[0..i-1]
int[] leftIn = Arrays.copyOfRange(in, 0, i);
// 构建当前根节点的右子树时, 前序遍历的数组为pre[i+1..pre.length-1]
int[] rightPre = Arrays.copyOfRange(pre, i + 1, pre.length);
// 构建当前根节点的右子树时, 中序遍历的数组为in[i+1..in.length-1]
int[] rightIn = Arrays.copyOfRange(in, i + 1, in.length);
// 递归构建当前根节点的左子树
root.left = reConstructBinaryTree(leftPre, leftIn);
// 递归构建当前根节点的右子树
root.right = reConstructBinaryTree(rightPre, rightIn);
// 递归构建二叉树完成, 返回根节点
return root;
}
TreeNode数据结构:
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) {
val = x;
}
}
来源:oschina
链接:https://my.oschina.net/u/2503731/blog/666620