236.二叉樹最近公共祖先

給定一個二叉樹, 找到該樹中兩個指定節點的最近公共祖先。

百度百科中最近公共祖先的定義爲:“對於有根樹 T 的兩個結點 p、q,最近公共祖先表示爲一個結點 x,滿足 x 是 p、q 的祖先且 x 的深度儘可能大(一個節點也可以是它自己的祖先)。”

例如,給定如下二叉樹:  root = [3,5,1,6,2,0,8,null,null,7,4]

示例 1:

輸入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
輸出: 3
解釋: 節點 5 和節點 1 的最近公共祖先是節點 3。
示例 2:

輸入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
輸出: 5
解釋: 節點 5 和節點 4 的最近公共祖先是節點 5。因爲根據定義最近公共祖先節點可以爲節點本身。
 

說明:

所有節點的值都是唯一的。
p、q 爲不同節點且均存在於給定的二叉樹中。

來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/er-cha-shu-de-zui-jin-gong-gong-zu-xian-lcof
著作權歸領釦網絡所有。商業轉載請聯繫官方授權,非商業轉載請註明出處。

思路:

  • 遞歸
  • 終止條件:
    • 當越過葉節點,則直接返回 nullnull ;
    • 當 rootroot 等於 p, qp,q ,則直接返回 rootroot ;
  • 遞推工作:
    • 開啓遞歸左子節點,返回值記爲 leftleft ;
    • 開啓遞歸右子節點,返回值記爲 rightright ;
  • 返回值: 根據 leftleft 和 rightright ,可展開爲四種情況;
    • 當 leftleft 和 rightright 同時爲空 :說明 rootroot 的左 / 右子樹中都不包含 p,qp,q ,返回 nullnull ;
    • 當 leftleft 和 rightright 同時不爲空 :說明 p, qp,q 分列在 rootroot 的 異側 (分別在 左 / 右子樹),因此 rootroot 爲最近公共祖先,返回 rootroot ;
    • 當 leftleft 爲空 ,rightright 不爲空 :p,qp,q 都不在 rootroot 的左子樹中,直接返回 rightright 。具體可分爲兩種情況:
      • p,qp,q 其中一個在 rootroot 的 右子樹 中,此時 rightright 指向 pp(假設爲 pp );
      • p,qp,q 兩節點都在 rootroot 的 右子樹 中,此時的 rightright 指向 最近公共祖先節點 ;
    • 當 leftleft 不爲空 , rightright 爲空 :與情況 3. 同理;
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
    if (root == null || p == root || q == root) { // 能夠判斷是根節點的情況
      return root;
    }
    TreeNode leftRoot = lowestCommonAncestor(root.left, p, q); // 假設兩個節點在左子樹上
    TreeNode rightRoot = lowestCommonAncestor(root.right, p, q); // 假設兩個節點在右子樹上
    if (leftRoot == null) return rightRoot;
    if (rightRoot == null) return leftRoot;
    return root;
  }

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章