题解 | #二叉搜索树的最近公共祖先#
二叉搜索树的最近公共祖先
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; } };