数据结构与算法--2.数组的定位排序

问题：

• 给定一个数组A以及下标i，将数组元素进行调整，使得所有比A[i]小的元素排在前边，接着是所有等于A[i]的元素，最后排列的是比A[i]大的元素

思路：

1. 第一步：将数组分成两部分，一部分元素都小于pivot，第二部分都大于或者等于pivot；
2. 第二步：将第二部分元素再次分成两部分，第一部分所有元素都等于pivot，第二部分元素都大于pivot.

def rearrangeByPivot(array, begin, end, pivot, checkEqual):

    if end<=begin:
        return

    while 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] = temp
            end -= 1
        else:
            begin += 1
    return array

def rearrangeArray(array,i):

    if (len(array)<=1):
        return array

    pivot = 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 array

S = [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]