Leetcode：236. 二叉树的最近公共祖先（C++）

目录

问题描述：

实现代码与解析：

原理思路:

---

问题描述：

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

百度百科中最近公共祖先的定义为：“对于有根树 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

实现代码与解析：

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) 
    {
        if(root==NULL) return NULL;
        if(root==p) return p;
        if(root==q) return q;
        TreeNode* left=lowestCommonAncestor(root->left,p,q);
        TreeNode* right=lowestCommonAncestor(root->right,p,q);
        //找到公共祖先返回
        if(left==p&&right==q||left==q&&right==p) return root;
        //已经找到祖先，就返回祖先,只找到一个结点，返回其此结点，两个结点都未找到，返回NULL
        return left?left:right;    
    }   
};

原理思路:

        比较简单啊，就直接写解题过程了，这里用的是递归。

        1、首先确定终止条件，很明显，当我们遇到了NULL或者p，q这两个结点就返回。

if(root==NULL) return NULL;
if(root==p) return p;
if(root==q) return q;

        2、然后就是递归逻辑，很明显，这题需要左右子树的返回信息，所以用后序遍历来处理。

TreeNode* left=lowestCommonAncestor(root->left,p,q);
TreeNode* right=lowestCommonAncestor(root->right,p,q);
.........
//后序处理代码

        3、之后就是后序的处理代码了，当该结点的左子树返回了q结点，右子树返回了p结点或者左右换一下顺序，左子树返回了p结点，右子树返回了q结点，我们就返回该根结点。

//找到公共祖先返回
if(left==p&&right==q||left==q&&right==p) return root;

        4、最后这步是重点了，虽然代码简单，但是其很精妙，处理了很多种情况，先把代码放出来，然后再给出解答。

return left?left:right;

第一种情况：

        q，p当前均未找到时，这行代码会返回NULL，符合逻辑。

第二种情况：

        q，p当前只找到了一个时，这行代码返回非空的（也就当前找到的那个），符合逻辑。

第三种情况：

        q，p的祖先已经找到了，当前结点的左右子树有一颗返回了祖先，有一颗返回了NULL，这行代码返回非空的（也就是祖先），符合逻辑。

第四种情况：

        q为p子树的某个结点或p为q子树的某个结点，这时，q或p自己就为其祖先，由终止条件可知，此情况下，遇到祖先就返回了，祖先的根节点，就符合第三种情况，返回了祖先，符合逻辑。

        所以别看这行代码短，但是却为这题的关键所在，大家在写题时也要把所有情况都考虑到，思考如何用最简单的代码来表达出来。