236. 二叉树的最近公共祖先 - 力扣(LeetCode) (leetcode-cn.com)
二叉树的最近公共祖先
递归
- 自底向上
- 回溯
- 后序遍历
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
// 自底向上 == 回溯 == 后序遍历
if (!root || root == p || root == q) return root;
TreeNode* left = lowestCommonAncestor(root->left, p, q);
TreeNode* right = lowestCommonAncestor(root->right, p, q);
// 如果left都和right是p或q的祖先,那么root就是最小公共祖先
if (left && right) return root;
// 如果left为空,那么p或q在right内, right是p或q的一个祖先
else if (!left && right) return right;
// 如果right为空,那么p或q在left内, left是p或q的一个祖先
else if (left && !right) return left;
// 否则找不到pq祖先
else return nullptr;
}
};
235. 二叉搜索树的最近公共祖先 - 力扣(LeetCode) (leetcode-cn.com)
二叉搜索树的最近公共祖先
递归
-
BST性质使得祖先值和子孙值有关系利用这层关系可以快速查找
-
自顶向下,我们遇到的第一个值在pq之间的结点就是最近公共祖先。
-
先序遍历
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
//p, q 都在左子树,到左子树处理
if (root->val > p->val && root->val > q->val)
return lowestCommonAncestor(root->left, p, q);
//p, q 都在右子树,到右子树处理
else if (root->val < p->val && root->val < q->val)
return lowestCommonAncestor(root->right, p, q);
//p, q 分别在左右子树中,那么返回此时结点
else return root;
}
};
非递归
- 自顶向下,我们遇到的第一个值在pq之间的结点就是最近公共祖先。
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
while (root) {
//p, q 都在左子树,到左子树处理
if (root->val > p->val && root->val > q->val) root = root->left;
//p, q 都在右子树,到右子树处理
else if (root->val < p->val && root->val < q->val) root = root->right;
//p, q 分别在左右子树中,那么返回此时结点
else return root;
}
return nullptr;
}
};