分治算法寻找第K小数

最新推荐文章于 2022-04-28 10:58:16 发布

Jarvi-W

最新推荐文章于 2022-04-28 10:58:16 发布

阅读量2k

点赞数 1

分类专栏： LeetCode 文章标签：分治算法

本文链接：https://blog.csdn.net/ace_15/article/details/77995775

版权

本文介绍了如何使用分治算法来解决找到数组中第K小元素的问题。首先讨论了经典问题"Selection"，并指出直接排序的低效性。接着，详细解释了分治策略，通过选取参考数将数组分为小于、等于和大于三部分，并递归地处理子数组。文章还提到了LeetCode上的378. Kth Smallest Element in a Sorted Matrix问题，这是一个在有序矩阵中找第K小元素的变体，利用了矩阵的有序特性结合二分查找来优化解决方案，采用了非递归的方法实现。

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

经典问题

有一道经典的算法题”Selection”

SELECTION
Input: A list of numbers S; an integer k
Output: The kth smallest element of S
For instance, if k = 1, the minimum of S is sought, whereas if k = |S|/2, it is the median.

这道题有很多解法，最简单的就是先对所有数排序。但是显然当数据较大时，这种方法效率不高。使用分治算法就可以在此基础上提高效率，因为只需对一部分而不是全部数据进行排序。
大致思路如下：每一次，先从数组中选出任意一个参考数pivot。选取的方法几乎不会影响效率，一般来说随机选取即可。选取好pivot后，对数组进行遍历，将所有数据分为三组，即小于，等于，大于pivot的三组。例如：