地址:
https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree/
描述:
思路:
(1) 如果当前结点 root 等于 NULL,则直接返回 NULL
(2) 如果 root 等于 p 或者 q ,那这棵树一定返回 p 或者 q
(3) 然后递归左右子树,因为是递归,使用函数后可认为左右子树已经算出结果,用 left 和 right 表示
(4) 此时若left为空,那最终结果只要看 right;若 right 为空,那最终结果只要看 left
(5) 如果 left 和 right 都非空,因为只给了 p 和 q 两个结点,都非空,说明一边一个,因此 root 是他们的最近公共祖先
(6) 如果 left和 right 都为空,则返回空(其实已经包含在前面的情况中了)
公共祖先是两个节点的祖先,祖先可以是一个节点的祖先
lowestCommonAncestor这个函数不要理解为找公共祖先,而就理解为帮两个节点找祖先 传入的值是root, p, q,帮p和q找到一个祖先就行,找到两个就更好了,如果找不到就返回NULL 在root->left里面找一次,root->right里面再找一次,如果某一边返回值是NULL, 那么说明两个值都在另一边 由于找的时候,一定是找的最近的祖先返回,所以这里直接返回前面的返回值就行了,可以保证是最近的公共祖先 如果左右的返回值都不是NULL,那说明p和q分别在两边,则当前节点就是最近公共祖先 左右都找不到就直接返回NULL
代码:
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(!root||root == p || root == q)
return root;
TreeNode* left = lowestCommonAncestor(root->left, p, q);
TreeNode* right = lowestCommonAncestor(root->right, p, q);
if(left == NULL)
return right;
if(right == NULL)
return left;
return root;
}
};