Leetcode 230. Kth Smallest Element in a BST

本文介绍了一种在二叉搜索树中查找第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]

参考链接


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值