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

/**
 * 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;
    }
};