quick select算法

最新推荐文章于 2024-10-26 08:00:00 发布

转载最新推荐文章于 2024-10-26 08:00:00 发布 · 1.1k 阅读

本文介绍了一种类似于快速排序的算法——Quickselect算法。该算法的主要特点是每次递归只需处理一个子数组，从而使得查找第k小元素的时间复杂度降低到O(n)。文章通过示例代码展示了如何使用Quickselect算法实现这一过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

本文的主角.quick select算法.其实就类似于快排.不同地方在于quick select每趟只需要往一个方向走.

时间复杂度:O(n).

def qselect(A,k):  
    if len(A)<=k:return A  
    pivot = A[-1]  
    right = [pivot] + [x for x in A[:-1] if x>=pivot]  
    rlen = len(right)  
    if rlen==k:  
        return right  
    if rlen>k:  
        return qselect(right, k)  
    else:  
        left = [x for x in A[:-1] if x<pivot]  
        return qselect(left, k-rlen) + right  
  
for i in range(1, 10):  
    print qselect([11,8,4,1,5,2,7,9], i)