二叉搜索树中的中序后继

Question

这是一个常见的算法问题。我试图在二叉搜索树中找到中序后继。这是我的代码

def inorderSuccessor(self, root, p):
        """
        :type root: TreeNode
        :type p: TreeNode
        :rtype: TreeNode
        """
        # if Node has a right child, go all the way down
        if p.right:
            curr = p
            while curr.left:
                curr = curr.left
            return curr
        # first ancestor whose left child the node is
        ans = None
        curr = root
        while curr is not p:
            if curr.val < p.val:
                curr = curr.right
            else:
                ans = curr
                curr = curr.left
        return ans

问题是当 p 是树中的最高节点时，这不起作用。我一直在为如何让这个边缘案例运转而摸不着头脑。

Answer 1

您可以在 BST 类中实现如下方法

def InOrderSucc(self,childNode):
    if(self.search(childNode)):
        if not (self.data == childNode):
            return(self.right.InOrderSucc(childNode))
        else:
            if(self.right):
                return(self.right.data)
            else:
                return None
    else:
        return False

其中 childNode 是您请求查找其 InOrder Successor 的 Node，注意 InOrder Successor 将始终是正确的 Child。