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

迭代中序遍历从最小的节点开始计数,返回第k个就是第K小的元素

def kthSmallest(self, root, k):
        stack = []
        while root or stack:
            if root:
                stack.append(root)
                root = root.left
            else:
                tmp = stack.pop()
                k -= 1
                if k == 0:
                    return tmp.val
                root = tmp.right

递归中序遍历,和迭代思路一样先找到最小的然后开始计数。返回第k个就是第k小的

def kthSmallest(self, root, k):
        self.res = 0
        self.count = 0
        def inorder(root):
            if not root:
                return 
            inorder(root.left)
            self.count += 1
            if self.count == k:
                self.res = root.val
                return
            inorder(root.right)
        inorder(root)
        return self.res
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值