问题：
- 给定一个数组A以及下标i，将数组元素进行调整，使得所有比A[i]小的元素排在前边，接着是所有等于A[i]的元素，最后排列的是比A[i]大的元素
思路：
- 第一步：将数组分成两部分，一部分元素都小于pivot，第二部分都大于或者等于pivot；
- 第二步：将第二部分元素再次分成两部分，第一部分所有元素都等于pivot，第二部分元素都大于pivot.
def rearrangeByPivot(array, begin, end, pivot, checkEqual):if end<=begin:returnwhile begin < end:# 如果checkEqual为真，那么交换条件是大于等于；如果是假，则元素交换条件为大于if (checkEqual is True and array[begin] >= pivot) or (checkEqual is False and array[begin] > pivot):#交换array[again]和array[end]temp = array[begin]array[begin] = array[end]array[end] = tempend -= 1else:begin += 1return arraydef rearrangeArray(array,i):if (len(array)<=1):return arraypivot = array[i]# 思路第一步代码：array = rearrangeByPivot(array,0,len(array)-1,pivot,True)# 找到第一部分与第二部分的分界点，确定下标j的值for j in range(len(array)):if array[j] >= pivot:break# 执行代码第二步：array =rearrangeByPivot(array, j, len(array)-1, pivot, False)return arrayS = [6,5,5,7,9,4,3,3,4,6,8,4,7,9,2,1]
i = 5
S = rearrangeArray(S,i)
print(S)
- 运行结果
[1, 2, 3, 3, 4, 4, 4, 6, 8, 7, 7, 9, 5, 5, 6, 9]
