二叉树的最近公共祖先

在这里插入图片描述

  • 假设我们需要找到结点p,q的最近公共祖先,我们可以发现二叉树的最近公共祖先满足:此结点的左结点和右节点分别存在p,q或者此节点恰好是p,q之一并且他的左右结点之一恰好是另一结点。
  • 我们假设fleft==true表示左节点或左节点的子节点中存在其中一个元素,fleft == false则表示左节点或左节点的子节点中不存在元素,fright同理,则判断一个结点是否最近公共祖先结点的判断条件是:

( f l e f t (fleft (fleft&& f r i g h t ) ∣ ∣ ( ( x = p ) ∣ ∣ ( x = q ) fright)||((x=p)||(x=q) fright)((x=p)(x=q)&& ( f l e f t ∣ ∣ f r i g h t ) ) (fleft||fright)) (fleftfright))

  • 第一个条件就是说这个结点满足左右节点分别存在p,q。第二个条件就是说满足此节点其中之一满足等于两个结点之一并且他的子节点包含另一个结点。
  • 并且我们是自底向上从叶子节点开始更新的,所以在所有满足条件的公共祖先中一定是深度最大的祖先先被访问到
  • 只要某个结点他的左右子结点之一满足等于p或q,我们就将此结点的bool值定位true。
TreeNode* ans;
bool dfs(TreeNode* root,TreeNode* p,TreeNode *q){
    if(root==nullptr) return false;
    bool fleft=dfs(root->left,p,q); //判断fleft
    bool fright=dfs(root->right,p,q);
    if((fleft&&fright)||((root->val==q->val||(root->val==p->val))&&(fleft||fright))) ans=root;
    return fleft||fright||(root->val==q->val||(root->val==p->val)); //满足其一就返回true
}
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
    dfs(root,p,q);
    return ans;
}

也可以用以下方法计算二叉树的最近公共祖先:

TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == nullptr || p == root || q == root) {
            return root;
        }

        TreeNode* l = lowestCommonAncestor(root->left, p, q);
        TreeNode* r = lowestCommonAncestor(root->right, p, q);
    
        return l == nullptr ? r : (r == nullptr ? l : root);
    }
  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值