二叉树力扣刷题（5）236. 二叉树的最近公共祖先

持续创作，加速成长！这是我参与「掘金日新计划 · 6 月更文挑战」的第9天，点击查看活动详情

236. 二叉树的最近公共祖先

难度中等1783

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

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

 

示例 1：

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

示例 2：

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

示例 3：

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

 

提示：

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

思路：

后序遍历，依次返回最近公共祖先，空节点和非空节点的最近公共祖先是非空节点，非空节点和非空节点的最近公共祖先是其节点。 如下代码 if (left && right) return root; if (left == NULL && right) return right; if (left && right == NULL) return left;

代码

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == q || root == p || root == NULL) return root;         //如果找到了节点p或者q，或者空节点，那么就返回这个节点
        TreeNode*left = lowestCommonAncestor(root->left, p, q);          //搜索整棵树
        TreeNode*right = lowestCommonAncestor(root->right, p, q);
        if (left && right) return root;                                  //都不为空说明节点就是root
        if (left == NULL && right != NULL) return right;                 //若有一个不为空，则返回那个节点
        else if (left && right == NULL) return left;
        else return NULL;
    }
};