重建二叉树

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如输入前序遍历序列{1, 2, 4, 7, 3, 5, 6, 8}和中序遍历序列{4, 7, 2, 1, 5, 3, 8, 6},则重建出其二叉树并输出它的头结点。

前序遍历:中左右
中序编列:左中右
后续编列:左右中
->点我

思路

在前序排序遍历序列中,第一个总是数的根节点,但在中序编序列中,根节点的值在序列的中间,
在中序编序列中,左子树的节点的值位于根节点值的左边,右子树的节点的值位于根节点的右边

实现

public TreeNode reConstructBinaryTree(int[] pre, int[] in) {
	return reBuild(pre, in, 0, pre.length - 1, 0, in.length - 1);
}
private TreeNode reBuild(int[] pre, int[] in, int preHead, int preEnd, int inHead, int inEnd) {
	TreeNode node = new TreeNode(pre[preHead]);
	if (preHead > preEnd || inHead > inEnd) {//出口
		return null;
	}

	int index = inHead;
	while (node.val != in[index] && index <= inEnd) {
		index++;
	}
	if (index > inEnd) {
		throw new RuntimeException("数组不符合规范!");
	}
	int leftLength = index - inHead;
	if (leftLength > 0) {//有 左子树
		node.left = reBuild(pre, in, preHead + 1, preHead + leftLength,
				inHead, index - 1);
	}
	if (leftLength < inEnd - inHead) {//有 右子树
		node.right = reBuild(pre, in, preHead + leftLength + 1, preEnd, index + 1, inEnd);
	}
	return node;
}

测试用例

public void test() {
	int[] pre = { 1, 2, 4, 7, 3, 5, 6, 8 };
	int[] in = { 4, 7, 2, 1, 5, 3, 8, 6 };
	TreeNode root = reConstructBinaryTree(pre, in);
	System.out.print("test:");
	preOrderTraverse(root);
	System.out.print("//");
	inOrderTraverse(root);
	System.out.println();
}
private void preOrderTraverse(TreeNode node) {
	if (node == null)
		return;
	System.out.print(node.val);
	preOrderTraverse(node.left);
	preOrderTraverse(node.right);
}
private void inOrderTraverse(TreeNode node) {
	if (node == null)
		return;
	inOrderTraverse(node.left);
	System.out.print(node.val);
	inOrderTraverse(node.right);
}

输出

test:12473568//47215386
posted @ 2019-09-06 14:48  tangmeng  阅读(268)  评论(0编辑  收藏  举报