剑指offer 68 - I. 二叉搜索树的最近公共祖先

剑指 Offer 68 - I. 二叉搜索树的最近公共祖先 - 力扣（LeetCode） (leetcode-cn.com)

目录

题解

运行结果

拓展：查找任意二叉树的任意两个节点的最近公共祖先 

思路

代码（含注释）

检验运行结果

---

题解

由于树是二叉搜索树，根据其特点直接查找即可

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        while (root) {
            if (root->val > p->val && root->val > q->val) {
                root = root->left;
            }
            else if (root->val < p->val && root->val < q->val) {
                root = root->right;
            }
            else return root;
        }
        return NULL;
    }
};

运行结果

拓展：查找任意二叉树的任意两个节点的最近公共祖先 

思路

寻找二叉树中任意两个节点的最近公共祖先，可以首先导出从根节点到这两个节点的路径，再比较两条路径，两条路径上重合的最后一个节点即为它们的最近公共祖先。

于是问题的重点变成了如何导出从根节点到一个节点的路径。如果我们用非递归后序遍历的方式来查找二叉树中的节点，那么找到该节点时，栈中的节点序列正是从根节点到该节点的路径。于是我们只需对非递归后序遍历二叉树的代码稍作修改即可。

代码（含注释）

#define MIN_SIZE 0x1F
class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        vector<TreeNode*> path1, path2;
        bool flag = true;    //用来记录已经找到几条路径

        TreeNode* nodeStack[MIN_SIZE];    //存储节点的栈
        bool state[MIN_SIZE];        //与节点栈对应的，存储节点状态的栈，true表示该节点的右子节点没有被访问过，false表示该节点的右子节点已被访问过
        size_t buoy = 0;    //栈顶指针
        while (root || buoy) {
            while (root) {
                nodeStack[buoy] = root;
                state[buoy] = true;
                ++buoy;
                root = root->left;
            }
            while (buoy) {
                root = nodeStack[buoy - 1];
                bool& Rside_unprocessed = state[buoy - 1];
                if (Rside_unprocessed) {
                    root = root->right;
                    Rside_unprocessed = false;
                    break;
                }
                if (root == p || root == q) {
                    if (flag) {    //找到第一条路径
                        path1 = vector<TreeNode*>(nodeStack, nodeStack + buoy);
                        flag = false;
                    }
                    else {    //找到第二条路径
                        path2 = vector<TreeNode*>(nodeStack, nodeStack + buoy);
                        goto CMP_PATH;    退出查找过程，转去处理两条路径
                    }
                }
                root = NULL;
                --buoy;
            }
        }
    CMP_PATH:
        TreeNode* res = NULL;
        for (size_t i = 0; i < min(path1.size(), path2.size()); ++i) {
            if (path1[i] == path2[i]) res = path1[i];
            else break;
        }
        return res;
    }
};

检验运行结果