基础练习——快速排序

快排采用分治递归策略,从待排序数组选一个基准值,比如首元素,然后剩下的元素里,比基准值小的放左边,大的放右边。这两部分继续递归进行排序。

首元素做基准

def quick_sort(lst):
    length = len(lst)
    if length <= 1:
        return lst

    key = lst[0]                                                # 选第一个作为基准值
    left = [lst[i] for i in range(1, length) if lst[i] < key]   # 比基准值小的放左边
    right = [lst[i] for i in range(1, length) if lst[i] >= key] # 比基准值大的放右边

    return quick_sort(left) + [key] + quick_sort(right)   

末元素做基准

也可以选择末元素做基准值,这样可以pop删掉再循环,代码简洁了些。当然也可以选择中位值作为基准值。

def quick_sort(lst):
    length = len(lst)
    if length <= 1:
        return lst

    key = lst[-1]                        # 选最后一个作为基准值
    lst.pop()
    left = [i for i in lst if i < key]   # 比基准值小的放左边
    right = [i for i in lst if i >= key] # 比基准值大的放右边

    return quick_sort(left) + [key] + quick_sort(right)     

优化比较次数

上面的代码一目了然,比较pythonic,但是在划分左右的时候比较次数重复了,比基准值小的没必要再去比较是否比基准值大,可以优化。

    key = lst[0]
    left, right = [], []
    for i in range(1, length):
        if lst[i] < key:
            left.append(lst[i])
        else:
            right.append(lst[i])

原地调整数组

之前的代码都开辟新数组,能否直接原地inplace排序?这是我们见得比较多的快排的一个版本,设置左右指针往中间靠拢。

def quick_sort(lst, left=0, right=None):
    if not left:
        left = 0
    if not right:
        right = len(lst) - 1
    if right - left < 1:
        return

    # 保存分治的边界
    low = left
    high = right

    key = lst[left]                                 # 左值作为基准保存
    while left < right:
        while lst[right] >= key and left < right:   # 从右边开始找第一个比基准值小的,注意,必须先看右边的,因为左值是基准!
            right -= 1
        lst[left] = lst[right]                      # 右值赋值给左值, 首位(左)、右分别为 小、小
        
        while lst[left] <= key and left < right:    # 从左边开始找第一个比基准值大的
            left += 1
        lst[right] = lst[left]                      # 左值赋值给右值,至此,左右互换一次,首位、左、右分别为 小、大、大
    else:
        lst[left] = key                             # 左右相撞,赋值为基准值,首位、左、右分别为 小、基、大 
        quick_sort(lst, low, left - 1)
        quick_sort(lst, left + 1, high)

其实这个代码我觉得有点拗口,可以理解为基准值往中间冒泡。

非递归版本

采用堆栈

def quick_sort(lst):
    if len(lst) <= 1:
        return lst

    stack = []
    stack.append(len(lst) - 1)
    stack.append(0)

    while stack:
        left = stack.pop()
        right = stack.pop()
        low = left
        high = right

        key = lst[left]
        while left < right:
            while lst[right] >= key and left < right:
                right -= 1
            lst[left] = lst[right]
            
            while lst[left] <= key and left < right:
                left += 1
            lst[right] = lst[left]

        else:
            lst[left] = key
            if low < left - 1:
                stack.append(left - 1)
                stack.append(low)
            if high > left + 1:
                stack.append(high)
                stack.append(left + 1)

时间复杂度

如果每次都能二等分,属于最优,那么就是O(nlgn);最差情况是每次都没分开,那么就是O(n²)。另外,平均时间复杂度也是O(nlgn),证明比较复杂。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,245评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,749评论 3 391
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,960评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,575评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,668评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,670评论 1 294
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,664评论 3 415
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,422评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,864评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,178评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,340评论 1 344
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,015评论 5 340
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,646评论 3 323
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,265评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,494评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,261评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,206评论 2 352

推荐阅读更多精彩内容

  • 知 识 点 / 超 人 数据结构算法排序是比较枯燥的知识,学习一定要耐着性子看,不然很容易理解错误。本文比较适合自...
    树下敲代码的超人阅读 5,236评论 9 58
  • 前言 查找和排序算法是算法的入门知识,其经典思想可以用于很多算法当中。因为其实现代码较短,应用较常见。所以在面试中...
    宝塔山上的猫阅读 1,086评论 1 21
  • 我是正宗如假包换的80后,正经历着要生二胎的抉择。 生活里遇到的同事都奔二胎去了,当然比较积极的多数是第一胎生女孩...
    杰克帅阅读 282评论 0 2
  • 继新年集五福游戏之后,马老板这个月又发明了一个新游戏:扫码得红包。扫的人得到红包,被扫的也得红包。真是双赢。加上,...
    懒懒的阿朱阅读 69评论 0 0