求BST中第K个最小的元素

最新推荐文章于 2022-07-05 16:05:42 发布

Storm-Shadow

最新推荐文章于 2022-07-05 16:05:42 发布

阅读量468

点赞数

分类专栏：剑指offer-算法与数据结构文章标签：算法数据结构 BST 二叉树

本文链接：https://blog.csdn.net/runninglion/article/details/107381287

版权

剑指offer-算法与数据结构专栏收录该内容

130 篇文章 246 订阅 ¥9.90 ¥99.00

订阅专栏

超级会员免费看

在给定的二叉搜索树(BST)中，寻找升序序列中的第k小元素。本文介绍了两种方法：一是利用中序遍历和栈辅助找到第k个元素；二是通过扩展BST节点结构，记录左子树节点数，实现O(h)复杂度的查找算法。

摘要由CSDN通过智能技术生成

给定一棵 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( pCra