题解 | #二叉搜索树的最近公共祖先#

二叉搜索树的最近公共祖先

https://www.nowcoder.com/practice/d9820119321945f588ed6a26f0a6991f

/**
 * struct TreeNode {
 *  int val;
 *  struct TreeNode *left;
 *  struct TreeNode *right;
 *  TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
class Solution {
  private:
    TreeNode* findNode(TreeNode* root, int val) {
        while (root) {
            if (val < root->val) {
                root = root->left;
            } else if (val > root->val) {
                root = root->right;
            } else {
                return root;
            }
        }
        return nullptr; // 如果没有找到,返回 nullptr
    }

    TreeNode* lowestCommonAncestorNode(TreeNode* root, int p, int q) {
        if (!root) return nullptr;

        TreeNode* pNode = findNode(root, p);
        TreeNode* qNode = findNode(root, q);

        // 确保找到了两个节点
        if (!pNode || !qNode) return nullptr;

        // 保证 p 的值小于 q 的值,简化逻辑
        if (pNode->val > qNode->val) swap(pNode, qNode);

        while (root) {
            if (root->val < pNode->val) {
                // 如果当前节点值小于 p 和 q 的值,向右子树移动
                root = root->right;
            } else if (root->val > qNode->val) {
                // 如果当前节点值大于 p 和 q 的值,向左子树移动
                root = root->left;
            } else {
                // 找到 LCA
                return root;
            }
        }
        return nullptr;
    }

  public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param root TreeNode类
     * @param p int整型
     * @param q int整型
     * @return int整型
     */
    int lowestCommonAncestor(TreeNode* root, int p, int q) {
        // write code here
        TreeNode * LCANode = lowestCommonAncestorNode(root, p, q);
        if (!LCANode) return -1;
        return LCANode->val;
    }
};

全部评论

相关推荐

杨柳哥:这不是普通人,那这个钱的是天才
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务