给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。
解题思路:两种情况:1.p q节点在不同子树,那么最近那个分岔路口就是共同祖先。2.q或p是另外一个节点的祖先 采用后续遍历,首先默认所有节点都是false,然后遍历叶子节点,把p/q两个节点染成true,再把所有子节点存在true的节点染成true,如果一个节点他的左右节点都是true,或者节点本身是p/q,子节点存在true,那他就是最近祖先
class Solution {
public:
vector<int> qval;
vector<int> pval;
TreeNode* ans;
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
dfs(root,p,q);
return ans;
}
bool dfs(TreeNode* root,TreeNode* p , TreeNode* q)
{
if(root == nullptr) return false;
bool l = dfs(root->left,p,q);
bool r = dfs(root->right,p,q);
if((l&&r) || ((root->val == p->val || root->val == q->val)&&(l || r)))
{
ans = root;
}
return l || r || (root->val == q->val || root->val == p->val);
}
};