第 K 個最小的快速排序 python

Question

我必須得到未排序數組中的第 K 個最小元素。 不是對整個數組進行排序，我只是嘗試對包含第 K 個元素的子數組進行排序。 然后我打印從 0 到 len(array) 的所有第 K 個

array = [6,5,4,3,2,1]
def quick_sort(lst, k):
    if len(lst) <= 1:
        return lst
    else:
        p = (lst[0] + lst[len(lst)-1])/2
        left = [x for x in lst if x <= p]
        right = [x for x in lst if x > p]
        
        if  k > len(left) -1 :
            k = k - len(left)+1
            return quick_sort(right,k)
        else:
            return quick_sort(left, k)
  
for i in range(len(array)):    
    print(*quick_sort(array,i+1))

我想得到 1、2、3、4、5、6，但我的代碼得到 2、3、5、6、6、6。 我需要改變什么？

PS 主要思想是不對所有 arrays 進行排序，也不使用 python 排序函數

Answer 1

array = [6, 5, 4, 3, 2, 1]

def quick_sort(lst, k):
    if len(lst) <= 1:
        return lst
    else:
        p = (lst[0] + lst[-1]) / 2
        left = [x for x in lst if x <= p]
        right = [x for x in lst if x > p]

        if k > len(left):
            k = k - len(left)
            return quick_sort(right, k)
        else:
            return quick_sort(left, k)

for i in range(len(array)):
    print(*quick_sort(array, i + 1))