剑指 Offer 54. 二叉搜索树的第k大节点

最新推荐文章于 2022-07-01 09:13:53 发布

原创最新推荐文章于 2022-07-01 09:13:53 发布 · 86 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

56 篇文章

订阅专栏

本文讲解如何使用递归实现二叉搜索树中第k大的节点查找，通过`findk`函数遍历树结构，找到第k个节点的值。适合理解搜索树数据结构和递归算法的应用。

难度简单188

给定一棵二叉搜索树，请找出其中第k大的节点。

示例 1:

输入: root = [3,1,4,null,2], k = 1
   3
  / \
 1   4
  \
   2
输出: 4

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

class Solution:
    def kthLargest(self, root: TreeNode, k: int) -> int:
        self.num=0

        def findk(cur):
            if not cur:
                return 
            right=findk(cur.right)
            if right:
                return right
            self.num+=1
            if k==self.num:
                return cur.val
            left=findk(cur.left)
            if left:
                return left
        return findk(root)