LeetCode 230. 二叉搜索树中第K小的元素

最新推荐文章于 2024-07-09 13:30:51 发布

镇长1998

最新推荐文章于 2024-07-09 13:30:51 发布

阅读量119

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode 暑假算法

本文链接：https://blog.youkuaiyun.com/weixin_41514525/article/details/97238285

Leetcode 同时被 2 个专栏收录

202 篇文章

订阅专栏

暑假算法

127 篇文章

订阅专栏

本文介绍了一个在二叉搜索树中查找第k个最小元素的算法。通过中序遍历将树节点值保存到数组中，利用BST性质使数组保持有序，从而快速定位目标元素。

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

给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。

说明：
你可以假设 k 总是有效的，1 ≤ k ≤ 二叉搜索树元素个数。

示例 1:

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

BST的中序遍历的应用，中序之后将所有结点的值保存在一个数组里，由于中序和BST的性质，数组本身就是有序的。

class Solution:
    
    def find_kth(self,root):
        if not root:
            return 
        
        self.find_kth(root.left)   
        self.res.append(root.val)
        self.find_kth(root.right)    
    
    def kthSmallest(self, root: TreeNode, k: int) -> int:
        self.res = []
        self.find_kth(root)
        return self.res[k-1]