一日一练:二叉搜索树的最近公共祖先

113 阅读1分钟

给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。

因为是二叉搜索树,所以两个节点的公共祖先节点会满足: 祖先节点的值大于较小节点的值且小于较大节点的值。又因为一个节点也可以是他自己的祖先节点,所以值相等也是符合的。

递归查找

function lowestCommonAncestor(
  root: TreeNode | null,
  p: TreeNode,
  q: TreeNode
): TreeNode | null {
  if (!root) return null

  if (
    // p为较小值节点,q为较大值节点
    (root.val >= p.val && root.val <= q.val) ||
    // p为较大值节点,q为较小值节点
    (root.val <= p.val && root.val >= q.val)
  ) {
    return root
  }
  return (
    lowestCommonAncestor(root.left, p, q) ||
    lowestCommonAncestor(root.right, p, q)
  )
}