考虑用递归函数的思路来做:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q)
这个函数从root为根节点的二叉树,寻找p和q的最近公共祖先,如果最近公共祖先存在返回root
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(root==NULL) return NULL;
if(root==p||root==q) return root;
TreeNode* left = lowestCommonAncestor(root->left,p,q);
TreeNode* right = lowestCommonAncestor(root->right,p,q);
if(left!=NULL&&right!=NULL) return root;
else if(left==NULL) return right;
else return left;
}
};
当我们用递归去做这个题时不要被题目误导,应该要明确一点
这个函数的功能有三个:给定两个节点 pp 和 qq
如果 pp 和 qq 都存在,则返回它们的公共祖先;
如果只存在一个,则返回存在的一个;
如果 pp 和 qq 都不存在,则返回NULL
本题说给定的两个节点都存在,那自然还是能用上面的函数来解决