分治算法求解列表中第k小的数

最新推荐文章于 2023-12-14 13:50:16 发布

酸辣螺丝粉

最新推荐文章于 2023-12-14 13:50:16 发布

阅读量2.8k

点赞数 1

分类专栏： python基本操作文章标签：求列表第k小的数分治算法

本文链接：https://blog.csdn.net/zhylhy520/article/details/80789601

版权

本文介绍了如何运用分治算法解决找到列表中第k小数的问题，避免了暴力排序带来的高时间复杂度。通过找到支点进行序列划分，根据k与支点索引的关系进行递归调用来定位第k小的数，实现O(n)的时间复杂度。

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

分治算法地思想就是将复杂问题分解为简单的子问题，然后寻求子问题的地归结，并组合各个子问题的解一起得到最终复杂问题的解。

针对求解列表中第k小的数，暴力拆解法可以将列表排序然后根据索引求出列表中第k小的数，显然这种算法的时间复杂度较大，下面介绍一下分治算法求列表中第k小的数，时间复杂度为O（n）.

Blum对于这个问题的解法就是通过找到一个被称为支点的数来对输入序列划分。支点左边的数都比支点数小，支点右边的数都比它大。如果k等于支点数的索引，这是可以立刻返回结果；如果k小于支点数索引，说明第k小的数在支点数左边，需要递归调用函数；如果k大于支点数索引，说明第k小的数在支点数右边，需要递归调用函数。

代码如下：

#寻找第K小的数
def select_fct(array,k):
    if len(array)<=10:#边界条件
        array=sorted(array)
        return array[k]
    pivot=get_pivot(array)  #得到支点数
    array_lt,array_gt,array_eq=patition_array(array,pivot)#按照支点数划分