LeetCode每日一题——面试题 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

题目

根据二叉搜索树的性质，大于p节点的节点往左边找，小于等于p节点的节点往右边找。

题解

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def inorderSuccessor(self, root: TreeNode, p: TreeNode) -> TreeNode:
        if root is None:
            return None
        if root.val <= p.val:
            return self.inorderSuccessor(root.right, p)
        
        node = self.inorderSuccessor(root.left, p)
        return root if node is None else node