力扣:235. 二叉搜索树的最近公共祖先
代码随想录
题目:
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉搜索树: root = [6,2,8,0,4,7,9,null,null,3,5]
思路:
因为是二叉搜索树,而且题目说明一定存在节点,只需要搜查每一个节点,然后判断此节点是否大于等于给定的小节点的值,小于等于给定的大节点的值,若满足则此节点即为所求。在搜查节点的时候,有方法的去寻找,如果当前节点值更大则搜索左树,如果当前节点值更小则搜索右子树。
代码:
class Solution {
public:
TreeNode* find(TreeNode* cur, TreeNode*p,TreeNode*q){
if(!cur) return NULL;
if(cur->val < p->val && cur->val < q->val) {
TreeNode* result = find(cur->right, p, q);
if(result) return result;
}
if(cur->val > p->val && cur->val > q->val) {
TreeNode* result = find(cur->left, p, q);
if(result) return result;
}
return cur;
}
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
return find(root, q, p);
}
};
迭代方法:
此迭代只不过是递归的一种while写法而已。不过比递归写法简单的多,而且易思考。
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;
}
};