重建二叉树
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如输入前序遍历序列{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