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

    一、题目

    求树中两个结点的最低公共祖先,此树不是二叉树,并且没有指向父节点的指针。

    二、解题思路

    我们首先得到一条从根结点到树中某一结点的路径,这就要求在遍历的时候,有一个辅助内存来保存路径.

    48.树中两个结点的最低公共结点 - 图1

    比如我们用前序遍历的方法来得到从根结点到H 的路径的过程是这样的:( 1 )遍历到A,把A 存放到路径中去,路径中只有一个结点A; ( 2 )遍历到B,把B 存到路径中去,此时路径为A->B; ( 3 )遍历到D,把D 存放到路径中去,此,时路径为A->B->D; ( 4 ) 遍历到F,把F 存放到路径中去,此时路径为A->B->D->F;( 5) F 已经没有子结点了,因此这条路径不可能到这结点H. 把F 从路径中删除,变成A->B->D; ( 6 )遍历G. 和结点F 一样,这条路径也不能到达H. 边历完G 之后,路径仍然是A->B->D; ( 7 )由于D 的所有子结点都遍历过了,不可能到这结点H,因此D 不在从A 到H 的路径中,把D 从路径中删除,变成A->B; ( 8 )遥历E,把E 加入到路径中,此时路径变成A->B->E, ( 9 )遍历H,已经到达目标给点, A->B->E 就是从根结点开始到达H 必须经过的路径。

    同样,我们也可以得到从根结点开始到达F 必须经过的路径是A->B。接着,我们求出这两个路径的最后公共结点,也就是B. B这个结点也是F 和H 的最低公共祖先.

    为了得到从根结点开始到输入的两个结点的两条路径,需要遍历两次树,每边历一次的时间复杂度是O(n)。

    三、解题代码

    1. public class Test{
    2. /**
    3. * 树的结点定义
    4. */
    5. private static class TreeNode {
    6. int val;
    7. List<TreeNode> children = new LinkedList<>();
    8. public TreeNode() {
    9. }
    10. public TreeNode(int val) {
    11. this.val = val;
    12. }
    13. @Override
    14. public String toString() {
    15. return val + "";
    16. }
    17. }
    18. /**
    19. * 找结点的路径
    20. *
    21. * @param root 根结点
    22. * @param target 目标结点
    23. * @param path 从根结点到目标结点的路径
    24. */
    25. public static void getNodePath(TreeNode root, TreeNode target, List<TreeNode> path) {
    26. if (root == null) {
    27. return;
    28. }
    29. // 添加当前结点
    30. path.add(root);
    31. List<TreeNode> children = root.children;
    32. // 处理子结点
    33. for (TreeNode node : children) {
    34. if (node == target) {
    35. path.add(node);
    36. return;
    37. } else {
    38. getNodePath(node, target, path);
    39. }
    40. }
    41. // 现场还原
    42. path.remove(path.size() - 1);
    43. }
    44. /**
    45. * 找两个路径中的最后一个共同的结点
    46. *
    47. * @param p1 路径1
    48. * @param p2 路径2
    49. * @return 共同的结点,没有返回null
    50. */
    51. public static TreeNode getLastCommonNode(List<TreeNode> p1, List<TreeNode> p2) {
    52. Iterator<TreeNode> ite1 = p1.iterator();
    53. Iterator<TreeNode> ite2 = p2.iterator();
    54. TreeNode last = null;
    55. while (ite1.hasNext() && ite2.hasNext()) {
    56. TreeNode tmp = ite1.next();
    57. if (tmp == ite2.next()) {
    58. last = tmp;
    59. }
    60. }
    61. return last;
    62. }
    63. /**
    64. * 找树中两个结点的最低公共祖先
    65. * @param root 树的根结点
    66. * @param p1 结点1
    67. * @param p2 结点2
    68. * @return 公共结点,没有返回null
    69. */
    70. public static TreeNode getLastCommonParent(TreeNode root, TreeNode p1, TreeNode p2) {
    71. if (root == null || p1 == null || p2 == null) {
    72. return null;
    73. }
    74. List<TreeNode> path1 = new LinkedList<>();
    75. getNodePath(root, p1, path1);
    76. List<TreeNode> path2 = new LinkedList<>();
    77. getNodePath(root, p2, path2);
    78. return getLastCommonNode(path1, path2);
    79. }
    80. }