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

最新推荐文章于 2024-07-14 14:55:48 发布

岁月标记

最新推荐文章于 2024-07-14 14:55:48 发布

阅读量205

点赞数 3

文章标签： leetcode 算法

本文链接：https://blog.csdn.net/akdjfhx/article/details/140378334

版权

给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。最近公共祖先的定义为：“对于有根树 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

/**
 * 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 {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if(!root) return nullptr;//节点不存在返回nullptr
        if(root==p) return p;//带回p
        if(root==q) return q;//带回q
        TreeNode* left;
        TreeNode* right;
        left=lowestCommonAncestor(root->left,p,q);
        right=lowestCommonAncestor(root->right,p,q);
        //回溯
        if(left&&right) return root;//pq在左右子树，返回当前节点
        if(!left&&right) return right;//带回right
        if(left&&!right) return left;//带回left
        return nullptr;    
    }

};