1. 程式人生 > 其它 >[Leetcode]35. 二叉搜尋樹的最近公共祖先

[Leetcode]35. 二叉搜尋樹的最近公共祖先

題目:

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

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

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

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

示例 2:

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

說明:
所有節點的值都是唯一的。
p、q 為不同節點且均存在於給定的二叉搜尋樹中。 思想:注意題目給出的是一個二叉搜尋樹,我們可以根據大小來遍歷分別找到兩個目標結點來記錄他們的路徑。同樣,也可以一次遍歷,從根節點出發,如果當前節點的值比兩個目標節點的值都大的話,那麼將當前節點指向左孩子,如果小,指向右孩子。如果一大一小,那麼說明遇到了分叉點,目標結點要麼是當前節點的孩子,要麼是子樹的一個節點,直接返回當前結點即可。
func lowestCommonAncestor(root, p, q *TreeNode) *TreeNode {
    var ans *TreeNode = root
    for {
        if ans.Val>p.Val&&ans.Val>q.Val{
            ans = ans.Left
        }else if ans.Val<p.Val&&ans.Val<q.Val{
            ans = ans.Right
        }else{
            return ans
        }
    }
}

題目來源:https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-search-tree