【LeetCode 235.二叉搜索树的最近公共祖先】JavaScript/C++实现(递归)

725 阅读2分钟

这是我参与8月更文挑战的第28天,活动详情查看:8月更文挑战


说明:文章部分内容及图片出自网络,如有侵权请与我本人联系(主页有公众号:小攻城狮学前端)

作者:小只前端攻城狮、 主页:小只前端攻城狮的主页、 来源:掘金

GitHub:P-J27、 CSDN:PJ想做前端攻城狮

著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。


【LeetCode 235.二叉搜索树的最近公共祖先】JavaScript/C++实现(递归)

题目描述:给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。

说明

  1. 所有节点的值都是唯一的。
  2. p、q 为不同节点且均存在于给定的二叉搜索树中。

例如,给定如下二叉搜索树: root = [6,2,8,0,4,7,9,null,null,3,5]

img

示例:

输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 4 输出: 2 解释: 节点 2 和节点 4 的最近公共祖先是 2, 因为根据定义最近公共祖先节点可以为节点本身。


思路分析

只要理清递归中过程即可。注意这是一棵二叉搜索树,满足 node.left.val < node.val < node.right.val

一共 4 种情况:

  1. 如果 p 和 q 的节点在 root 的两侧,那么最近公共祖先节点就是 root
  2. 如果 p.val 或者 q.val 等于 root.val,那么最近公共祖先节点就是 p 或 q
  3. 如果情况 1 和 2 都不满足,说明 p 和 q 在 root 的同一侧:
    • 若 root.val 大于 p.val 和 q.val,那么往左侧递归
    • 若 root.val 小于 p.val 和 q.val,那么往右侧递归

解法 1: JavaScript 实现


/**
 * @param {TreeNode} root
 * @param {TreeNode} p
 * @param {TreeNode} q
 * @return {TreeNode}
 */
var lowestCommonAncestor = function(root, p, q) {
  if (
    (root.val > p.val && root.val < q.val) ||
    (root.val < p.val && root.val > q.val) ||
    root.val === p.val ||
    root.val === q.val
  ) {
    return root;
  } else if (root.val > q.val) {
    return lowestCommonAncestor(root.left, p, q);
  } else {
    return lowestCommonAncestor(root.right, p, q);
  }
};

解法 2: C++实现

由于这题的剑指 offer 的题库中没提供 js 语言环境,我用 c++实现了思路:


class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (
            (root->val == p->val)
            || (root->val == q->val)
            || (root->val > q->val && root->val < p->val)
            || (root->val < q->val && root->val > p->val)
        ) {
            return root;
        } else if (root->val > q->val) {
            return lowestCommonAncestor(root->left, p, q);
        } else {
            return lowestCommonAncestor(root->right, p, q);
        }
    }
};

感谢阅读,希望能对你有所帮助,文章若有错误或者侵权,可以在评论区留言或在我的主页添加公众号联系我。

写作不易,如果觉得不错,可以「点赞」+「评论」 谢谢支持❤