重构二叉树之前序遍历和中序遍历

我是研究僧i 提交于 2019-12-02 07:16:12

题目描述:


输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。

假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

例如输入前序遍历序列{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;
	}
}


易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!