设计一个算法,找出二叉搜索树中指定节点的“下一个”节点(也即中序后继)。
如果指定节点没有对应的“下一个”节点,则返回null。
示例 1:
输入: root = [2,1,3], p = 1``
2
/
1 3
输出: 2
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/successor-lcci
# 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:
res = None
cur = root
while cur:
if cur.val <= p.val:
#遍历右孩子
cur = cur.right
else:
#记录前一个节点,如果所给节点没有右孩子,res就是所求
res = cur
cur = cur.left
return res