开启我的LeetCode刷题日记：面试题 04.06. 后继者

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

编程世界总是离不了算法

最近在看框架源码时，会有很多算法的实现逻辑，有时候会感到吃力

于是决定蹭着假期，加强算法和数据结构相关的知识

那怎么提升呢？

其实我知道算法这东西没有捷径，多写多练才能提升，于是我开启我的LeetCode刷题之旅

第一阶段目标是：200道，每天1到2篇

为了不乱，本系列文章目录分为三部分：

1. 今日题目：xxx
2. 我的思路
3. 代码实现

今日题目：面试题 04.06. 后继者

设计一个算法，找出二叉搜索树中指定节点的“下一个”节点（也即中序后继）。

如果指定节点没有对应的“下一个”节点，则返回null。

示例 1:

输入: root = [2,1,3], p = 1

  2
 / \
1   3

输出: 2

示例 2:

输入: root = [5,3,6,2,4,null,null,1], p = 6

      5
     / \
    3   6
   / \
  2   4
 /   
1

输出: null

来源：力扣（LeetCode） 链接：leetcode.cn/problems/su… 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

我的思路

1. 使用递归的中序遍历
2. 定义pre:上一个节点，res:目标节点的后继者
3. 遍历到当前节点时，将pre设置为当前节点值
4. 若上个节点pre等于p，那么当前节点就是后继者了，赋值给res
5. 注意递归出口：!root或者res有值，因为已经找到了后继者，直接返回即可

代码实现

const inorderSuccessor = (root, p) => {
    // pre:上一个节点
    // res:目标节点的后继者
    let [pre, res] = [null, null];
    const inOrder = root => {
        // res有值，就不必递归了
        if (!root || res) return;
        inOrder(root.left);
        // 上个节点pre等于指定节点p
        // 那么后继者就是当前节点root了
        if (pre === p) {
            res = root;
        }
        // 将pre设置为当前节点值
        pre = root;
        inOrder(root.right);
    };
    inOrder(root);
    return res;
};

总结

实现方式其实有很多，这里仅供参考~

由于刚开始刷题，也不知道从哪里刷好，如果前辈们有好的建议，希望不吝赐教，感谢🌹