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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

后序遍历的思想，先遍历子树再遍历根节点

若p在root的子树中，或p=root，则说root是p的祖先
寻找p,q的公共祖先时有三种情况:
1、p,q都在root的一侧（左侧或右侧），此时继续往左侧和右侧遍历
2、p=root,且q在root子树中，此时公共祖先是p
3、q=root，且p在root子树中，此时公共祖先是q

定义TreeNode* dfs(TreeNode *root,TreeNode *p,TreeNode *q)为p,q的最近公共祖先
所以首先检查root，if(root=q||root=p)说明最近公共祖先就是为root，如果不满足则检查左子树和右子树所以 left = dfs(root->left,p,q); right = dfs(root->right,p,q);

如果left!=NULL&&right!=NULL 说明p,q同时在root的子树中，所以root，为最近的公共祖先
如果left!=NULL&&right == NULL则说明p,q都不在右子树中，都在左子树中
如果right!=NULL&&left == NULL则说明p,q都不在左子树中，都在右子树中

代码

/**
 * 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) {
        return dfs(root,p,q);
    }
    TreeNode* dfs(TreeNode *root,TreeNode *p,TreeNode *q){
        if(root == nullptr) return nullptr;
        if(root == q || root == p) return root;
        TreeNode *left = dfs(root->left,p,q);
        TreeNode *right = dfs(root->right,p,q);
        if(left&&right) return root;
        if(left) return left;
        if(right) return right;
        return nullptr;
    }
};