算法导论第9章选择算法

waiting不是违停

于 2022-04-17 19:42:41 发布

阅读量177

点赞数

分类专栏：算法导论（Python）文章标签： python 算法

本文链接：https://blog.csdn.net/weixin_44737922/article/details/124234369

版权

算法导论（Python）专栏收录该内容

10 篇文章 0 订阅

订阅专栏

该博客介绍了如何运用基于快速排序的方法来选择列表中第i个最大的元素。通过RandomizedPartition函数进行随机划分，然后在RandomizedSelect函数中递归查找目标位置。代码示例展示了在Python中实现这一算法，并应用到一个具体的数组上，找出数组中第8大的数。

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

书p119
选择第i个最大的数，用基于快速排序的方法解决

import random
def Partitions(A

p,r):
    i = p-1
    x = A[r]
    for j in range(p,r):
        if A[j]<x:
            i+=1
            temp = A[i]
            A[i] = A[j]
            A[j] = temp
    temp = A[i+1]
    A[i+1] = A[r]
    A[r] = temp
    return i+1

def RandomizedPartition(A,p,r):
    i = random.randint(p,r)
    temp = A[i]
    A[i] = A[r]
    A[r] = temp
    return Partitions(A,p,r)

def RandomizedSelect(A,p,r,i):
    if p==r:
        return A[p]
    else:

        q = RandomizedPartition(A,p,r)
        k = q-p+1
        if k==i:
            return A[q]
        elif i<k:
            return RandomizedSelect(A,p,q-1,i)
        else:
            return RandomizedSelect(A, q+1, r, i-k)

A = [2,8,7,1,3,5,6,4,10]
print(RandomizedSelect(A,0,len(A)-1,8))