创见博客
二叉树的最近公共祖先
七崽爱吃小饼干2026/01/10阅读 0专栏 算法合集

二叉树的最近公共祖先

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

百度百科中最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”

示例 1:

codeType
输入:root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
输出:3
解释:节点 5 和节点 1 的最近公共祖先是节点 3 。

示例 2:

codeType
输入:root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
输出:5
解释:节点 5 和节点 4 的最近公共祖先是节点 5 。因为根据定义最近公共祖先节点可以为节点本身。

示例 3:

codeType
输入:root = [1,2], p = 1, q = 2
输出:1

提示:

  • 树中节点数目在范围 [2, 105] 内。
  • -109 <= Node.val <= 109
  • 所有 Node.val 互不相同 。
  • p != q
  • p 和 q 均存在于给定的二叉树中。

解法

ts
/**
 * Definition for a binary tree node.
 * class TreeNode {
 *     val: number
 *     left: TreeNode | null
 *     right: TreeNode | null
 *     constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
 *         this.val = (val===undefined ? 0 : val)
 *         this.left = (left===undefined ? null : left)
 *         this.right = (right===undefined ? null : right)
 *     }
 * }
 */

let ans: TreeNode | null = null

function lastOrder(root: TreeNode | null, p: TreeNode | null, q: TreeNode | null): boolean{
    if(!root) return false
    const l = lastOrder(root.left, p, q) // 左子树是否含目标
    const r = lastOrder(root.right, p, q) // 右子树是否含目标
    // 左右子树含目标 或者 本身就是目标之一且子树含目标
    if((l && r) || ((root.val === p.val || root.val === q.val) && (l || r))){
        ans = root
    }
    // 是否含目标或者本身就是目标
    return l || r || (root.val === p.val || root.val === q.val)
}

function lowestCommonAncestor(root: TreeNode | null, p: TreeNode | null, q: TreeNode | null): TreeNode | null {
	// 可以按照后序遍历来标记当前节点是不是目标节点的祖先
    lastOrder(root, p, q)
    return ans
};
评论
0/100