剑指offer面试题6 重建二叉树（java实现）

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

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;
	}

}