LCR 174. 寻找二叉搜索树中的目标节点 - 力扣(LeetCode)




要在二叉搜索树(BST)中找到第 cnt 大的员工编号,我们可以利用 中序遍历的逆序(右-根-左) 的特性。
解题思路:
- 逆序中序遍历(右-根-左):BST 右子树的值比根节点大,左子树的值比根节点小,所以逆序中序遍历可以按 降序 顺序访问节点。
- 计数:使用一个计数器,当遍历到第
cnt个节点时,返回该节点的值。
代码实现(Python):
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
class Solution:
def findKthLargest(self, root: TreeNode, cnt: int) -> int:
self.result = None
self.k = cnt
def reverse_inorder_traversal(node):
if not node or self.k == 0:
return
# 先遍历右子树
reverse_inorder_traversal(node.right)
# 访问当前节点
self.k -= 1
if self.k == 0:
self.result = node.val
return
# 遍历左子树
reverse_inorder_traversal(node.left)
reverse_inorder_traversal(root)
return self.result
复杂度分析:
- 时间复杂度:O(H + k),最坏情况 O(N)(退化成链表)。
- 空间复杂度:O(H),递归栈的深度,最坏情况 O(N)。
这个方法能够高效地找到二叉搜索树中的第 cnt 大的员工编号。
846

被折叠的 条评论
为什么被折叠?



