[递归/二叉树]LeetCode 236.二叉树的最近公共祖先

这篇博客讨论了如何解决LeetCode上的二叉树最近公共祖先问题。作者分享了两种递归解法,一种是通过设置标志变量来追踪目标节点,另一种是官方的解法,两者都避免了平方级的时间复杂度。博客强调了正确理解题目条件和利用递归在解决树结构问题中的重要性。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题面

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

百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
链接:https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree

题目分析

一开始想用并查集,然而发现递归挺好使的
注意到两个条件可以更新返回值且唯一:

  1. 某个节点左右两边都存在一个我们我找的p/q节点其中的一个,那么这个节点就是根节点,即我们要求的节点。因为树中没有相同val的节点所以可以这样使用而不用区分。
  2. 目标节点是其中另一个目标节点的最近公共祖先,这种情况下需要判断自己的值和遍历左右子树中是否有我们要寻找的p/q节点。
    当然还有其他想法,我最初的想法是这样的:
    首先树根节点肯定是所有节点的祖先,那么就先将ans置为root,然后判断根节点的左节点或者右节点是否能够更新为答案,即在左子树中是否有p\q节点,如果有则继续遍历左节点的左右节点,如果没有查看右子树中是否有p/q节点,如果有则继续更新答案。这样的话时间会远远超过上述的递归方法,因为这样的解法粗略的估计为O(N^2),而上述方法为O(N)
AC代码
第二种解法:
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
    TreeNode* ans;
    bool flag1;
    bool flag2;
public:
    bool recursive(TreeNode* root, TreeNode* p, TreeNode* q) {
        if(!root)
            return false;
        if(root->val == p->val) flag1 = true;
        if(root->val == q->val) flag2 = true;
        return (flag1 && flag2) || recursive(root->left, p, q) || recursive(root->right, p, q);
    }
    void getAns(TreeNode* root, TreeNode* p, TreeNode* q) {
        flag1 = flag2 = false;
        bool f1 = recursive(root->left, p, q);
        flag1 = flag2 = false;
        bool f2 = recursive(root->right, p, q);
        if(f1) 
            getAns(root->left, p, q);
        else if(f2)
            getAns(root->right, p, q);
        else {
            ans = root;
        }
        return;
    }
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        ans = root;
        getAns(root, p, q);
        return ans;
    }
};

第二种解法(官方解法)

class Solution {
public:
    TreeNode* ans;
    bool dfs(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == nullptr) return false;
        bool lson = dfs(root->left, p, q);
        bool rson = dfs(root->right, p, q);
        if ((lson && rson) || ((root->val == p->val || root->val == q->val) && (lson || rson))) {
            ans = root;
        } 
        return lson || rson || (root->val == p->val || root->val == q->val);
    }
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        dfs(root, p, q);
        return ans;
    }
};

本人水平有限,如有任何错误,恳请大家指正,不要误导他人…

错误反馈: 1004183196@qq.com

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值