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

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

题目描述

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

百度百科中最近公共祖先的定义为：“对于有根树 T 的两个结点 p、q，最近公共祖先表示为一个结点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”
例如，给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4]

示例 1:

示例 2:

说明:所有节点的值都是唯一的。p、q 为不同节点且均存在于给定的二叉树中。

思路

在左、右子树中分别查找是否包含p或q：
如果以下两种情况（左子树包含p，右子树包含q/左子树包含q，右子树包含p），那么此时的根节点就是最近公共祖先
如果左子树包含p和q，那么到root->left中继续查找，最近公共祖先在左子树里面
如果右子树包含p和q，那么到root->right中继续查找，最近公共祖先在右子树里面
1.此时若left为空，那最终结果只要看 right；若 right 为空，那最终结果只要看 left
2.如果 left 和 right 都非空，因为只给了 p 和 q 两个结点，都非空，说明一边一个，因此 root 是他们的最近公共祖先
3.如果 left 和 right 都为空，则返回空（其实已经包含在前面的情况中了）

代码

/**
 * 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 == NULL || root == p || root == q){
            return root;
        }
        TreeNode* left = lowestCommonAncestor(root->left,p,q);
        TreeNode* right = lowestCommonAncestor(root->right,p,q);
        return left == NULL ? right : (right == NULL ? left : root); 
    }
};

复杂度分析

时间复杂度：O(N)，N 是二叉树中的节点数，最坏情况下，我们需要访问二叉树的所有节点。
空间复杂度：O(N)，这是因为递归堆栈使用的最大空间为N,斜二叉树的高度可以是 N。