【每日一题】LeetCode. 236. 二叉树的最近公共祖先

每日一题，防止痴呆 = =

一、题目大意

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

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

例如，给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4]


来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree

二、题目思路以及AC代码

拿到题之后，直接想偏了，感觉我做到树的题就GG。

一开始我的思路是，DFS会把祖先存在栈里，然后我们只要在查到第一个数的时候回溯，并把祖先都标记，然后查到第二个数的时候，回溯，找到第一个被标记的数就可以了，后来发现，DFS好像是会存右节点，这就麻烦了，好像这样的思路和官解有一个有点像，反正这个想了半天没想出来，心态就崩了，就去找题解了。

思路一

我们可以定义一个函数f(t)，用来定义以t为根节点的树，是否包含p或q。那么当一个节点满足如下条件的时候，它就是p和q的最近公共祖先： f(lchild)&&f(rchild) || (t->val == p->val || t->val == q.val) && (f(lchild) || f(rchild))。

其中f(lchild)&&f(rchild)表示该根节点的左右子树中一个包含p，一个包含q，那么t为公共祖先，又由于进行递归调用，所以是从底向上遍历，所以是最近公共祖先。
后面的一个条件表示根节点是p或q，然后其左右子树中有一个包含另一个数，那么t也为公共祖先。

可能有人觉得会不会有一个根节点的子树包含p和q，那上面的条件就不成立了。但由于是自底向上进行遍历，所以第一个公共祖先肯定不是有一个子树包含两个的情况，可以仔细想一下。

思路二

还有一种思路可能更好理解，就是用哈希表去对每一个树节点索引到他的父节点，这个复杂度是O(n)，然后就可以先从p向上找，将其所有的父节点标记，然后再从q向上找，找到的第一个被标记的节点就是最近公共祖先。

其实我一开始想过这么做，也没仔细算复杂度，就是觉得很麻烦，就被我抛弃了 = =，还是题做的少。

下面给出思路一的AC代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */

bool dfs(TreeNode* root, TreeNode* p, TreeNode* q, TreeNode*& ans) {
    if (!root) return false;
    bool l = dfs(root->left, p, q, ans);
    bool r = dfs(root->right, p, q, ans);

    if ((l && r) || (p->val == root->val || q->val == root->val) && (l || r)) {
        ans = root;
    }
    return l || r || (root->val == p->val) || (root->val == q->val);
}

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        TreeNode* res;
        dfs(root, p, q, res);
        return res;
    }
};

如果有问题，欢迎大家指正！！！