【树-中等】面试题 04.06. 后继者

菜菜2022

于 2021-08-18 19:27:22 发布

阅读量82

点赞数

CC 4.0 BY-SA版权

分类专栏：刷题 # leetcode

本文链接：https://blog.youkuaiyun.com/kz_java/article/details/119787184

leetcode 同时被 2 个专栏收录

714 篇文章

订阅专栏

刷题

643 篇文章

订阅专栏

本文介绍如何设计算法，通过深度优先搜索找到二叉搜索树中指定节点的中序后继。提供两种方法，包括递归和迭代方式，以及示例代码解析。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

如果指定节点没有对应的“下一个”节点，则返回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
【代码】
在这里插入图片描述

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

class Solution:
    def dfs(self,root,p):
        if not root:
            return 
        self.dfs(root.left,p)
        if self.cnt==1:
            self.ans=root
            self.cnt+=1
        if root.val==p.val:
            self.cnt+=1
        self.dfs(root.right,p)
    def inorderSuccessor(self, root: TreeNode, p: TreeNode) -> TreeNode:
        self.ans=None
        self.cnt=0
        self.dfs(root,p)
        return self.ans

【方法2】
在这里插入图片描述

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