NC102 在二叉树中找到两个节点的最近公共祖先(c++)

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 * };
 */

class Solution {
public:
    /**
     * 
     * @param root TreeNode类 
     * @param o1 int整型 
     * @param o2 int整型 
     * @return int整型
     */
    int lowestCommonAncestor(TreeNode* root, int o1, int o2) {
        // write code here
        return dfs(root, o1, o2)->val;
        
    }
    
    TreeNode* dfs(TreeNode* root, int o1, int o2){
        if(root == nullptr || root->val == o1 || root->val == o2){
            return root;
        }
        TreeNode* left = dfs(root->left, o1, o2);
        TreeNode* right = dfs(root->right, o1, o2);
        //如果left为空，说明这两个节点在root结点的右子树上，我们只需要返回右子树查找的结果即可
        if(!left){
            return right;
        }
        //如果right为空，说明这两个节点在root结点的左子树上，我们只需要返回左子树查找的结果即可
        if(!right){
            return left;
        }
        //如果left和right都不为空，说明这两个节点一个在root的左子树上一个在root的右子树上，
        //我们只需要返回root结点即可。
        return root;
    }
};