「前端刷题」235.二叉搜索树的最近公共祖先(MEDIUM)

57 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第3天,点击查看活动详情

题目(Lowest Common Ancestor of a Binary Search Tree)

链接:https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-search-tree
解决数:1593
通过率:67.7%
标签:树 深度优先搜索 二叉搜索树 二叉树 
相关公司:amazon microsoft linkedin 

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

百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”

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

 

示例 1:

输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8
输出: 6 
解释: 节点 2 和节点 8 的最近公共祖先是 6

示例 2:

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

 

说明:

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

思路

如果树节点的值位于p和q两节点值中间或与某个节点相等,则该节点一定是两节点最近父节点 否则如果树节点值大于两节点值,则两节点都位于树节点的左子树,将其左节点作为当前节点进行搜索 同理如果树节点值小于两节点值,则两节点都位于树节点的右子树,将其右节点作为当前节点进行搜索

let lowestCommonAncestor = function(root, p, q) {
    let temp = root;
    if(p.val > q.val) {
        [p, q] = [q, p];
    }
    while(true) {
        if((temp.val > p.val && temp.val <= q.val) || (temp === p || temp === q)) {
            return temp;
        } else if(temp.val < p.val && temp.val < q.val) {
            temp = temp.right;
        } else {
            temp = temp.left;
        }
    }
};

先遍历找到其中一个节点,并用WeakSet存储遍历经历过的节点 再遍历查找第二个节点,如果set中存在这个点,更新公共点,但不跳出遍历,因为这不一定是最近。接着遍历,直到已找到目标点