剑指offer面试题6 重建二叉树(java实现)

来源:互联网 发布:python hmmlearn库 编辑:程序博客网 时间:2024/05/19 13:45

解题思路:已知二叉树的前序序列和中序序列,那么前序序列的第一个元素,就是根节点,此时在中序序列中遍历根节点对应的值,记下该值的索引,此时该索引左边元素属于左子树,可以确定左子树的长度以及左子树的前序、中序范围,索引的右边元素属于右子树,可以确定右子树的长度以及右子树的前序、中序范围。接下来可以用递归地方式不断地在左子树、右子树中进行上述操作。递归的终止条件是当前序和中序范围重叠,并且指向元素也相同,此时即可返回。

public class Solution {public class TreeNode {int val;TreeNode left;TreeNode right;TreeNode(int x) {val = x;}}public TreeNode reConstructBinaryTree(int[] pre, int[] in) {if (pre == null || in == null || pre.length == 0 || in.length == 0 || pre.length != in.length) {return null;}int startPre = 0;int endPre = pre.length - 1;int startIn = 0;int endIn = in.length - 1;return reConstructBinaryTreeCore(pre, in, startPre, endPre, startIn, endIn);}/** * @param pre *            前序序列 * @param in *            中序序列 * @param startPre *            前序序列起始位置 * @param endPre *            前序序列结束位置 * @param startIn *            中序序列起始位置 * @param endIn *            中序序列结束位置 * @return 前序序列范围代表一棵树或者是某棵子树 中序序列范围代表对应的那一棵树或者是某棵子树 */private TreeNode reConstructBinaryTreeCore(int[] pre, int[] in, int startPre, int endPre, int startIn, int endIn) {// 前序序列的第一个节点就是根节点TreeNode root = new TreeNode(pre[startPre]);// 递归终止的结束条件if (startPre == endPre) {if (startIn == endIn && pre[startPre] == in[startIn]) {return root;}}// 在中序序列中找到值为startPre的索引int indexOfRoot;for (indexOfRoot = startIn; indexOfRoot <= endIn; indexOfRoot++) {if (in[indexOfRoot] == pre[startPre]) {break;}}if (indexOfRoot == endIn - startIn + 1) {System.out.println("输入不合法");}// 此时的indexOfRoot指向中序遍历中根节点的位置// 接下来求出左子树的前序范围、中序范围,右子树的前序范围、中序范围// 求出左、右子树的长度// 左子树的长度int leftTreeLength = indexOfRoot - startIn;int rightTreeLength = endIn - indexOfRoot;if (leftTreeLength > 0) {// 对左子树进行一次建树操作root.left = reConstructBinaryTreeCore(pre, in, startPre + 1, startPre + leftTreeLength, startIn,indexOfRoot - 1);}if (leftTreeLength < endPre - startPre) {// 如果有右子树,对右子树进行一次建树操作root.right = reConstructBinaryTreeCore(pre, in, endPre - rightTreeLength + 1, endPre, indexOfRoot + 1,endIn);}return root;}}


原创粉丝点击