[Leetcode]35. 二叉搜尋樹的最近公共祖先
阿新 • • 發佈:2021-11-21
題目:
給定一個二叉搜尋樹, 找到該樹中兩個指定節點的最近公共祖先。百度百科中最近公共祖先的定義為:“對於有根樹 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