剑指 Offer 68 - II. 二叉树的最近公共祖先(简单)
题目:
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(root==p||root==q||root==nullptr) return root; //如果当前节点为空或者当前节点即为其中某个指定节点,则将其暂时视作公共祖先,返回该节点
TreeNode* left = lowestCommonAncestor(root->left, p, q); //在左子树中寻找p或q
TreeNode* right = lowestCommonAncestor(root->right, p, q); //在右子树中寻找p或q
if(left&&right) return root; //p和q分别在左右子树中
if(!left) return right; //p和q都在右子树中
return left; //p和q都在左子树中
}
};
以上代码转自代码随想录