数组的定位排序

最新推荐文章于 2023-02-26 14:10:37 发布

qq_33386711

最新推荐文章于 2023-02-26 14:10:37 发布

阅读量275

点赞数

分类专栏：面试刷题文章标签：算法数据结构 python

本文链接：https://blog.csdn.net/qq_33386711/article/details/104784838

版权

面试刷题专栏收录该内容

2 篇文章 0 订阅

订阅专栏

题目描述

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

A： = [6,5,5,7,9,4,3,3,4,6,8,4,7,9,2,1]
i = 5
A[i] = 4
A =[1, 2, 3, 3, 4, 4, 4, 6, 8, 7, 7, 9, 5, 5, 6, 9]

解题思路

两部走，
开始将小于A[i]的排序，
在将等于A[i]的排列

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[begin] 和 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]
    #先执行算法第一步骤，将数组元素分两部分，第一部分小于pivot,第二部分大于等于pivot
    array = rearrangeByPivot(array, 0, len(array)-1, pivot, True)
    #找到第一部分和第二部分的分界点
    for j in range(len(array)):
        if array[j] >= pivot:
            break
    #执行算法第二步骤
    array = rearrangeByPivot(array, j, len(array)-1, pivot, False)
    return array