leedcode 二叉搜索树中第k小元素

最新推荐文章于 2025-06-04 17:15:36 发布

futurewq

最新推荐文章于 2025-06-04 17:15:36 发布

阅读量109

点赞数

CC 4.0 BY-SA版权

分类专栏： leedcode

本文链接：https://blog.youkuaiyun.com/wangrongrongwq/article/details/99761591

leedcode 专栏收录该内容

50 篇文章

订阅专栏

本文介绍了一个在二叉搜索树中查找第K个最小元素的算法。通过中序遍历的方式，可以有效地找到目标元素。示例包括了具体的二叉搜索树结构和预期的输出结果。

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

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

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

示例 1:

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

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

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

class Solution(object):
    def kthSmallest(self, root, k):
        """
        :type root: TreeNode
        :type k: int
        :rtype: int
        """
        cnt = 0
        s = []
        p = root
        while p or len(s) > 0:
            while(p):
                s.append(p)
                p = p.left
            p = s[-1]
            s.pop(-1)
            cnt += 1
            if cnt == k:
                return p.val
            p = p.right