多轴快速排序

Question

def partition(A, k):
  # your code goes here
  pivots = insertionsort(A[-k:])
  arr = [[]]*(k+1)
  for el in A[:-k]:
    if el <= pivots[0]:
      arr[0].append(el)
    elif el >= pivots[-1]:
      arr[-1].append(el)
    else:
      for i in range(1, len(pivots) - 1):
        if pivots[i] <= el < pivots[i+1]:
          arr[i].append(el)
          break

我正在尝试使用k个枢轴进行快速排序。并且分区应该给出k + 1个部分。该功能的参数无法更改。请帮忙。