問題：
- 給定一個數組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]
