题解 | #农场牛群族谱#

农场牛群族谱

https://www.nowcoder.com/practice/7d28855a76784927b956baebeda31dc8

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @param p int整型 
     * @param q int整型 
     * @return int整型
     */
    TreeNode* traversal(TreeNode* root, int q, int p) {
        // 从左右去找q和p,只要找到一个就返回
        // 如果root 为空,或者root是q p 中的某一个值,当前递归的场景是在root孩子中找,
        // 所以如果root是q p中的某一个,则root必然是p q的共同的父节点
        if (root == nullptr || root->val == q || root->val == p) return root;
        // 如果root不是,则分别从左右去找
        TreeNode* left = traversal(root->left, q, p);
        TreeNode* right = traversal(root->right, q, p);
        if (!left) return right;
        if (!right) return left;
        // 如果左右都找到了,则表示当前节点为共同父节点
        return root;
    }
    int lowestCommonAncestor(TreeNode* root, int p, int q) {
        // write code here
        if (root == nullptr) return -1;
        return traversal(root, q, p)->val;
    }
};

全部评论

相关推荐

点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务