235. 二叉搜索树的最近公共祖先
?做题思路or感想:
-
递归一条边的写法:
if (dfs(root->left))return; if (dfs(root->right))return;
-
搜索整棵树的写法
left = dfs(root->left); right = dfs(root->right); //left与right的逻辑处理
在递归函数又返回值的情况下:
- 如果搜索一条边,那么在递归函数的返回值不为空的时候,立刻返回
- 如果搜索整棵树,则直接用变量
left
和right
保存返回值,接下来的逻辑就是用变量left
和right
来处理当前节点
方法一:就当作一颗普通二叉树来做
用后序遍历从底部开始往上遍历,因为是要找最近公共祖先,所以要搜索整棵树!
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
//递归中止条件
if (root == nullptr || root == p || root == q)return root;
//后序遍历
TreeNode* left = lowestCommonAncestor(root->left, p, q);
TreeNode* right = lowestCommonAncestor(root->right, p, q);
//单层逻辑:由左右子树递归的结果来判断该节点是不是最近公共祖先
//是,则返回值
if (left == p && right == q || left == q && right == p)return root;
//不是,则继续传递递归的信息
else if (!left && right)return right;
else if (left && !right)return left;
else return nullptr;
}
};
但这种方法完全没有用到二叉搜索树的特性,暴珍天物啊
方法二:利用二叉搜索树的性质
如果cur
节点的数值在[p, q]区间,则说明该节点cur就是最近公共祖先
明白上面的话后,就直接照做就好了
但是可以利用二叉搜索树的特性,如果节点值大于这个区间,就往左子树跑,如果小于这个区间,就往右子树跑
class Solution {
public:
bool flag = 1;
TreeNode* result = nullptr;
void dfs(TreeNode* cur, TreeNode* max, TreeNode* min) {
if (flag == 1) { //这个flag可有可无,我自己习惯加的,用来控制递归的进行与否
if (cur == nullptr)return; //递归中止条件,其实这里也是可有可无,因为不可能遍历到空节点的
if (cur->val <= max->val && cur->val >= min->val) { //这里注意是大于等于和小于等于,题目里有说祖先节点可以是那两个节点的本身,故是闭区间[p, q]
result = cur;
flag = 0;
return;
}
//根据二叉搜索树的特性
if (cur->val > max->val)dfs(cur->left, max, min);
if (cur->val < min->val)dfs(cur->right, max, min);
}
}
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if (q->val > p->val)swap(p, q); //细节,这里必须让p是大的那个!当然也可以让q是大的那个,随自己喜欢
dfs(root, p, q);
return result;
}
};