剑指 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;
}
};