求BST中第K个最小的元素

给定一棵 BST,请在这棵 BST 上查找其对应的升序序列中的第 k 个元素(即第 k 小元素)。如下图的 BST,如果 k=3,那么所求节点为 10,如果 k=5,那么所求节点为 14。

解法一:中序遍历法,我们用一个栈作为辅助进行中序遍历,在遍历过程中,对出栈次数进行计数,出栈 k 次时即找到我们要的节点。算法复杂度为 O(n),n 为树的节点总数,算法描述如下:

/* initialization */
pCrawl = root
set initial stack element as NULL (sentinal)

/* traverse upto left extreme */
while(pCrawl is valid )
   stack.push(pCrawl)
   pCrawl = pCrawl.left

/* process other nodes */
while( pCrawl = stack.pop() is valid )
   stop if sufficient number of elements are popped.
   if( pCrawl.right is valid )
      pCrawl = pCrawl.right
      while( pCrawl is valid )
         stack.push(pCrawl)
         pCrawl = pCrawl.left

C 实现


                
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
好的,这里给出一个基于序遍历的方法实现: ```python class TreeNode: def __init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right class Solution: def kthSmallest(self, root: TreeNode, k: int) -> int: self.k = k self.res = None self._dfs(root) return self.res def _dfs(self, root): if not root: return self._dfs(root.left) self.k -= 1 if self.k == 0: self.res = root.val return self._dfs(root.right) ``` 首先我们定义一个 `TreeNode` 类用于构建二叉搜索树。然后定义一个 `Solution` 类,其 `kthSmallest` 方法用于返回第 `k` 小的元素。这里我们使用成员变量 `self.k` 来记录还需要找的元素个数,初始值为 `k`。使用成员变量 `self.res` 来记录找到的第 `k` 小的元素。 接着定义一个 `_dfs` 方法用于递归遍历二叉搜索树。在遍历过程,我们先递归遍历左子树,然后将 `self.k` 减一,如果此时 `self.k` 等于零,说明已经找到了第 `k` 小的元素,将 `self.res` 赋值为当前节点的值,并且直接返回。否则递归遍历右子树。 最后在 `kthSmallest` 方法调用 `_dfs` 方法遍历整个二叉搜索树,并返回 `self.res` 即可。 时间复杂度为 $O(h+k)$,其 $h$ 为二叉搜索树的高度。因为我们在遍历的过程只会遍历到目标元素,所以时间复杂度与目标元素的深度有关。在最坏情况下,即二叉搜索树退化成一个链表时,时间复杂度为 $O(n)$,其 $n$ 为节点个数。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Storm-Shadow

你的鼓励将是我最大的动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值