题目描述:
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 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 lowestTree = new(TreeNode)
var dfsTree func(root,p,q *TreeNode)
dfsTree = func(root, p, q *TreeNode) {
if root.Val >= p.Val && root.Val <= q.Val || root.Val >= q.Val && root.Val <= p.Val{
lowestTree.Val = root.Val
return
}
if root.Val > p.Val && root.Val > q.Val{
dfsTree(root.Left,p,q)
}
if root.Val < p.Val && root.Val < q.Val{
dfsTree(root.Right,p,q)
}
}
dfsTree(root,p,q)
return lowestTree
}
官方优雅写法:
func lowestCommonAncestor(root, p, q *TreeNode) (ancestor *TreeNode) {
ancestor = root
for {
if p.Val < ancestor.Val && q.Val < ancestor.Val {
ancestor = ancestor.Left
} else if p.Val > ancestor.Val && q.Val > ancestor.Val {
ancestor = ancestor.Right
} else {
return
}
}
}
复杂度分析
时间复杂度:O(n),其中 nn 是给定的二叉搜索树中的节点个数。分析思路与方法一相同。
空间复杂度:O(1)。
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-search-tree/solution/er-cha-sou-suo-shu-de-zui-jin-gong-gong-zu-xian-26/