寻找数组中第K大的数字-快排解法(Python3)

给定一个数组A,寻找第k大的数字。

本质上是查询操作,但是却使用到快排思想(即分治)

A = [-1, 3, -2, 4, -5];

def partition(A: int, low: int, high: int) -> int:
    temp = A[low];
    while low < high:
        while low < high and temp >= A[high]:
            high -= 1;
        A[low] = A[high];
        while low < high and temp < A[low]:
            low += 1;
        A[high] = A[low];
    A[low] = temp;
    return low;

def quciksort(A: int, low: int, high: int, k: int) -> int:
    if low <= high:
        pos = partition(A, low, high);
        if pos == k - 1:
            return A[pos];
        elif pos > k - 1:
            return quciksort(A, low, pos-1, k);
        elif pos < k - 1:
            return quciksort(A, pos+1, high, k);
    else:
        return -1;

print(quciksort(A, 0, len(A)-1, 5));
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值