給定一個二叉搜索樹, 找到該樹中兩個指定節(jié)點的最近公共祖先键兜。
百度百科中最近公共祖先的定義為:“對于有根樹 T 的兩個結(jié)點 p凤类、q,最近公共祖先表示為一個結(jié)點 x普气,滿足 x 是 p谜疤、q 的祖先且 x 的深度盡可能大(一個節(jié)點也可以是它自己的祖先)∠志鳎”
例如夷磕,給定如下二叉搜索樹: root = [6,2,8,0,4,7,9,null,null,3,5]
image.png
示例 1:
輸入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8
輸出: 6
解釋: 節(jié)點 2 和節(jié)點 8 的最近公共祖先是 6。
示例 2:
輸入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 4
輸出: 2
解釋: 節(jié)點 2 和節(jié)點 4 的最近公共祖先是 2, 因為根據(jù)定義最近公共祖先節(jié)點可以為節(jié)點本身仔沿。
說明:
所有節(jié)點的值都是唯一的坐桩。
p、q 為不同節(jié)點且均存在于給定的二叉搜索樹中封锉。
/**
* Definition for TreeNode.
* type TreeNode struct {
* Val int
* Left *ListNode
* Right *ListNode
* }
*/
func lowestCommonAncestor(root, p, q *TreeNode) *TreeNode {
if root == nil || p == nil || q == nil {
return nil
}
for root != nil {
if p.Val > root.Val && q.Val > root.Val {
root = root.Right
} else if p.Val < root.Val && q.Val < root.Val {
root = root.Left
} else {
return root
}
}
return nil
}