• 一、题目
  • 二、解题思路
  • 三、解题代码

    一、题目

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

    二、解题思路

    由前序遍历的第一个节点可知根节点。根据根节点,可以将中序遍历划分成左右子树。在前序遍历中找出对应的左右子树,其第一个节点便是根节点的左右子节点。按照上述方式递归便可重建二叉树。

    三、解题代码

    1. public class Test {
    2. /**
    3. * 二叉树节点类
    4. */
    5. public static class BinaryTreeNode {
    6. int value;
    7. BinaryTreeNode left;
    8. BinaryTreeNode right;
    9. }
    10. /**
    11. * 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二节树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
    12. *
    13. * @param preorder 前序遍历
    14. * @param inorder 中序遍历
    15. * @return 树的根结点
    16. */
    17. public static BinaryTreeNode construct(int[] preorder, int[] inorder) {
    18. // 输入的合法性判断,两个数组都不能为空,并且都有数据,而且数据的数目相同
    19. if (preorder == null || inorder == null || preorder.length != inorder.length || inorder.length < 1) {
    20. return null;
    21. }
    22. return construct(preorder, 0, preorder.length - 1, inorder, 0, inorder.length - 1);
    23. }
    24. /**
    25. * 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二节树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
    26. *
    27. * @param preorder 前序遍历
    28. * @param ps 前序遍历的开始位置
    29. * @param pe 前序遍历的结束位置
    30. * @param inorder 中序遍历
    31. * @param is 中序遍历的开始位置
    32. * @param ie 中序遍历的结束位置
    33. * @return 树的根结点
    34. */
    35. public static BinaryTreeNode construct(int[] preorder, int ps, int pe, int[] inorder, int is, int ie) {
    36. // 开始位置大于结束位置说明已经没有需要处理的元素了
    37. if (ps > pe) {
    38. return null;
    39. }
    40. // 取前序遍历的第一个数字,就是当前的根结点
    41. int value = preorder[ps];
    42. int index = is;
    43. // 在中序遍历的数组中找根结点的位置
    44. while (index <= ie && inorder[index] != value) {
    45. index++;
    46. }
    47. // 如果在整个中序遍历的数组中没有找到,说明输入的参数是不合法的,抛出异常
    48. if (index > ie) {
    49. throw new RuntimeException("Invalid input");
    50. }
    51. // 创建当前的根结点,并且为结点赋值
    52. BinaryTreeNode node = new BinaryTreeNode();
    53. node.value = value;
    54. // 递归构建当前根结点的左子树,左子树的元素个数:index-is+1个
    55. // 左子树对应的前序遍历的位置在[ps+1, ps+index-is]
    56. // 左子树对应的中序遍历的位置在[is, index-1]
    57. node.left = construct(preorder, ps + 1, ps + index - is, inorder, is, index - 1);
    58. // 递归构建当前根结点的右子树,右子树的元素个数:ie-index个
    59. // 右子树对应的前序遍历的位置在[ps+index-is+1, pe]
    60. // 右子树对应的中序遍历的位置在[index+1, ie]
    61. node.right = construct(preorder, ps + index - is + 1, pe, inorder, index + 1, ie);
    62. // 返回创建的根结点
    63. return node;
    64. }
    65. }