主要思考如下几点:
- 如何自底向上遍历?
- 遍历整棵树,还是遍历局部树?
- 如何把结果传到根结点?
关于如果递归函数有返回值,如何区分是要搜索一条边,还是搜索整个树:
case1: 搜一条边的写法:
//若递归函数返回值不为空,则立即返回
if (递归函数(root->left)) return ;
if (递归函数(root->right)) return ;
case2: 搜索整个树的写法:
left = 递归函数(root->left);
right = 递归函数(root->right);
left与right的逻辑处理;
- 自底向上寻找,即回溯,二叉树回溯的过程即是自底向上,而后序遍历就是天然的回溯过程。
- 怎么判断一个结点是结点q和结点p的公共祖先呢:如果找到一个结点,其左子树出现结点q
,右子树出现结点p,或者反之,则,该结点即位结点p和q的最近公共祖先- 确定返回值类型:
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(root == q || root == p || root == NULL)
return root;
TreeNode* left = lowestCommonAncestor(root->left, p, q);
TreeNode* right = lowestCommonAncestor(root->right, p, q);
if(left != NULL && right != NULL) return root;
if(!left && right) return right;
else if(!right && left) return left;
else return NULL;
}
};
因为二叉搜索树是有序的,即可以告诉我们遍历方向,故可以使用前序遍历(这里什么遍历都可,因为不需要处理中间结点)
这里是遇到符合条件的结点就返回,故符合上面第一种情况
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
//if(root == NULL) return root;
//if(root->val >= p->val && root->val <= q->val)
//return root;
if(root->val > p->val && root->val > q->val){
TreeNode* l = lowestCommonAncestor(root->left, p, q);
if(l) return l;
}
if(root->val < p->val && root->val < q->val){
TreeNode* r = lowestCommonAncestor(root->right, p, q);
if(r) return r;
}
return root; //root节点在区间(p->val <= root->val && root->val <= root->val)或者 (root->val <= cur->val && cur->val <= root->val)中
}
};