一.二叉搜索树上查找最低公共祖先
输入的是两个值和二叉搜索树的根节点,输出最低公共祖先的值。
剑指offer上的题目:
给出递归和非递归的解法:
递归写法:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if (root == NULL) {
return root;
}
if (root->val < p->val && root->val < q->val) {
return lowestCommonAncestor(root->right, p, q);
}
if (root->val > p->val && root->val > q->val) {
return lowestCommonAncestor(root->left, p, q);
}
return root;
}
非递归写法:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
while (root) {
if (root->val < p->val && root->val < q->val) {
root = root->right;
} else if (root->val > p->val && root->val > q->val) {
root = root->left;
} else {
return root;
}
}
return NULL;
}
二.如果这个树不是二叉排序树,就是二叉树呢?
最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
如上图所示,D和F的最低公共祖先就是B。B和F的最低公共祖先也是B。
下面用递归方法解这道题。函数输入是根节点,节点1,节点2。
递归分三种情况return:
- 空树:直接return null
- 根节点等于其中一个节点,return根节点
- 在左子树和右子树中各自查找最低公共祖先,这里又分为三种情况:
如果在左子树中没找到(即左子树返回结果为null),return右子树的查找结果
如果在右子树中没找到(即右子树返回结果为null),return左子树的查找结果
否则,就直接返回根节点。
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if (root == NULL) {
return NULL;
}
if (root == p || root == q) {
return root;
}
TreeNode* left = lowestCommonAncestor(root->left, p, q);
TreeNode* right = lowestCommonAncestor(root->right, p, q);
if (left == NULL) {
return right;
}
if (right == NULL) {
return left;
}
return root;
}