Leetcode 230. Kth Smallest Element in a BST

最新推荐文章于 2021-09-03 20:56:56 发布

原创最新推荐文章于 2021-09-03 20:56:56 发布 · 296 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #中序遍历 #k #top #遍历

LeetCode 同时被 2 个专栏收录

90 篇文章

订阅专栏

二叉树

13 篇文章

订阅专栏

本文介绍了一种在二叉搜索树中查找第K小元素的方法，利用中序遍历的特性，通过迭代而非递归的方式找到目标值。这种方法不仅避免了递归带来的堆栈溢出风险，而且效率高，适用于各种规模的数据集。

题目描述：找出二叉搜索树的第k小的值。

题目链接：Leetcode 230. Kth Smallest Element in a BST

思路：二叉搜索树是严格的左子树小于根、根小于右子树，中序遍历结果为从小到大的有序列表。
故用中序遍历结果，当ans有k个的时候，返回最后一个值。
当然也可以用递归的方式来做。

代码如下

class Solution:
    def kthSmallest(self, root, k):
        """
        :type root: TreeNode
        :type k: int
        :rtype: int
        """
        #中序遍历结果 当为k的时候就可以返回值了。
        s = []
        curr = root
        ans = []
        while(s or curr):
            if curr:
                s.append(curr)
                curr = curr.left
            else:
                node = s.pop()
                ans.append(node.val)
                if len(ans)==k:
                    return ans[-1]
                curr = node.right
        return ans[k]