python - 在Python中实现快速排序

我正在尝试实现快速排序。看起来很简单，实现一个pivot函数，这样较小的元素和较大的元素就被收集在两个单独的列表中递归地这样做，直到列表足够小，可以在恒定时间内排序。

def pivot(a, pivot_index):
    # I guess you can keep two indexes one greater and one lesser and swap.
    i, j = 0, len(a)-2
    p = a[pivot_index]
    a[pivot_index] = a[len(a)-1]
    a[len(a)-1] = p

    while(i<j):
        if(a[i]<p):
            i = i + 1
        if(a[i]>p):
            temp = a[i]
            a[i] = a[j]
            a[j] = temp
            j = j - 1
        #print(a)
    return a[0:i], a[i:]

def quicksort(a):
    if len(a) <= 1:
        return a
    p = len(a)//2
    l1, l2 = pivot(a, p)
    return quicksort(l1) + quicksort(l2)

if __name__ == "__main__":
    a = [1,-9,288,91,3,4,58,67,8]
    print(quicksort(a))

我知道错误：

Traceback (most recent call last):
  File "11.py", line 79, in <module>
    print(quicksort(a))
  File "11.py", line 73, in quicksort
    return quicksort(l1) + quicksort(l2)
  File "11.py", line 73, in quicksort
    return quicksort(l1) + quicksort(l2)
  File "11.py", line 73, in quicksort
    return quicksort(l1) + quicksort(l2)
  [Previous line repeated 993 more times]

Traceback (most recent call last):
  File "11.py", line 76, in <module>
    print(quicksort(a))
  File "11.py", line 70, in quicksort
    return quicksort(l1) + quicksort(l2)
  File "11.py", line 70, in quicksort
    return quicksort(l1) + quicksort(l2)
  File "11.py", line 70, in quicksort
    return quicksort(l1) + quicksort(l2)
  [Previous line repeated 993 more times]
  File "11.py", line 69, in quicksort
    l1, l2 = pivot(a, p)
  File "11.py", line 54, in pivot
    while(i<j):
RecursionError: maximum recursion depth exceeded in comparison

可能发生的情况是，在quicksort中从未调用基本情况但不知道怎么解决。

最佳答案

下面的功能解释了快速排序的确切方式
通过将输入数据作为数组发送到quicksort函数来实现。
在这里，我们使用递归实现了快速排序，如下所示：
首先检查数组是否包含1个以上的元素如果数组中少于或等于一个元素，则返回数组。（如果len（arr）现在从数组的中间获取一个pivot元素（从数组大小的一半中检索元素），然后在第一次迭代中计算pivot元素（pivot=arr[len（arr）//2]）
如果数组中的元素小于pivot元素，则使用python中的list comprehension创建一个左数组（left=[x for x in arr if x如果数组中的元素等于pivot元素，则创建一个中间数组（middle=[x for x in arr If x==pivot]）
如果数组中的元素多于pivot元素，则创建一个右数组（right=[x for x in arr if x>pivot]）
将左数组、右数组发送到quicksort函数并递归地重复它们，直到数组中的所有元素都排序完毕。

 def quicksort(arr):
          if len(arr) <= 1:
             return arr
          pivot = arr[len(arr)//2]
          left = [x for x in arr if x < pivot]
          middle = [x for x in arr if x == pivot]
          right = [x for x in arr if x > pivot]
          return quicksort(left) + middle + quicksort(right)

关于python - 在Python中实现快速排序，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/48017422/