信息发布→ 登录 注册 退出

Python实现希尔排序,归并排序和桶排序的示例代码

发布时间:2026-01-11

点击量:
目录
  • 1. 前言
  • 2. 希尔排序
    • 2.1 前后切分
    • 2.2 增量切分
  • 3. 归并排序
    • 3.1 分解子问题
    • 3.2 求解子问题
    • 3.3 合并排序
  • 4. 基数排序
    • 5. 总结

      1. 前言

      本文将介绍希尔排序、归并排序、基数排序(桶排序)。

      在所有的排序算法中,冒泡、插入、选择属于相类似的排序算法,这类算法的共同点:通过不停地比较,再使用交换逻辑重新确定数据的位置。

      希尔、归并、快速排序算法也可归为同一类,它们的共同点都是建立在分治思想之上。把大问题分拆成小问题,解决所有小问题后,再合并每一个小问题的结果,最终得到对原始问题的解答。

      通俗而言:化整为零,各个击破。

      分治算法很有哲学蕴味:老祖宗所言 合久必分,分久必合,分开地目的是为了更好的合并。

      分治算法的求解流程:

      分解问题:将一个需要解决的、看起很复杂 原始问题 分拆成很多独立的子问题,子问题与原始问题有相似性。

      如:一个数列的局部(小问题)有序,必然会让数列最终(原始问题)有序。

      求解子问题:子问题除了与原始问题具有相似性,也具有独立性,即所有子问题都可以独立求解。

      合并子问题:合并每一个子问题的求解结果最终可以得到原始问题的解。

      下面通过深入了解希尔排序算法,看看分治算法是如何以哲学之美的方式工作的。

      2. 希尔排序

      讲解希尔之前,先要回顾一下插入排序。插入排序的平均时间复杂度,理论上而言和冒泡排序是一样的 O(n2),但如果数列是前部分有序,则每一轮只需比较一次,对于 n 个数字的原始数列而言,时间复杂度可以是达到 O(n)

      插入排序的时间复杂度为什么会出现如此有意思的变化?

      • 插入排序算法的排序思想是尽可能减少数字之间的交换次数。
      • 通常情形下,交换处理的时间大约是移动的 3 倍。这便是插入排序的性能有可能要优于冒泡排序的原因。

      希尔排序算法本质就是插入排序,或说是对插入排序的改良。

      其算法理念:让原始数列不断趋近于排序,从而降低插入排序的时间复杂度。

      希尔排序的实现流程:

      • 把原始数列从逻辑上切割成诸多个子数列。
      • 对每一个子数列使用插入排序算法排序。
      • 当所有子数列完成后,再对原数列进行最后一次插入算法排序。

      希尔排序算法的理念:当数列局部有序时,全局必然是趋向于有序”

      希尔排序的关键在于如何切分子数列,切分方式可以有 2 种:

      任何时候使用分治理念解决问题时,分拆子问题都是关键的也是核心的。

      2.1 前后切分

      如有原始数列=[3,9,8,1,6,5,7] 采用前后分成 2 个子数列。

      前后分算得上是简单粗暴的切分方案,没有太多技术含量,这种一根筋的切分方式,对于原始问题的最终性能优化可能起不了太多影响。

      如上图所示,对子数列排序后,如果要实现原始数列中的所有数字从小到大排列有序,则后部分的数字差不多全部要移到时前部分数字的中间,其移动量是非常大的。

      后面的 4 个数字中,1 需要移动 3 次,567 需要移动 2 次, 肉眼可见的次数是 9 次。

      这种分法很难实现数字局部有序的正态分布,因为数字的位置变化不大

      如下图是原始数列=[3,9,8,1,6,5,7] 的原始位置示意图:

      使用前后切分后的数字位置变化如下图所示,和上图相比较,数字的位置变化非常有限,而且是限定在一个很窄的范围内。也就是说子问题的求解结果对最终问题的结果的影响很微小。

      2.2 增量切分

      增量切分采用间隔切分方案,可能让数字局部有序以正态分布。

      增量切分,需要先设定一个增量值。如对原始数列=[3,9,8,1,6,5,7] 设置切分增量为 3 时,整个数列会被切分成 3 个逻辑子数列。增量数也决定最后能切分多少个子数列。

      对切分后的 3 个子数列排序后可得到下图:

      在此基础之上,再进行插入排序的的次数要少很多。

      使用增量切分后再排序,原始数列中的数字的位置变化范围较大。

      如数字 9 原始位置是 1,经过增量切分再排序后位置可以到 4。已经很接近 9 的最终位置 6 了。

      下图是增量切分后数字位置的变化图,可以看出来,几乎所有的数字都产生了位置变化 ,且位置变化的跨度较大。有整体趋于有序的势头。

      实现希尔排序算法时,最佳的方案是先初始化一个增量值,切分排序后再减少增量值,如此反复直到增量值等于 1 (也就是对原数列整体做插入排序)。

      增量值大,数字位置变化的跨度就大,增量值小,数字位置的变化会收紧。

      编码代码希尔排序:

      # 希尔排序
      def shell_sort(nums):
          # 增量
          increment = len(nums) // 2
          # 新数列
          while increment > 0:
              # 增量值是多少,则切分的子数列就有多少
              for start in range(increment):
                  insert_sort(nums, start, increment)
              # 修改增量值,直到增量值为 1    
              increment = increment // 2
              
      # 插入排序
      def insert_sort(nums, start, increment):
          for back_idx in range(start + increment, len(nums), increment):
              for front_idx in range(back_idx, 0, -increment):
                  if nums[front_idx] < nums[front_idx - increment]:
                      nums[front_idx], nums[front_idx - increment] = nums[front_idx - increment], nums[front_idx]
                  else:
                      break
      
      nums = [3, 9, 8, 1, 6, 5, 7]
      shell_sort(nums)
      print(nums)
      

      这里会有一个让人疑惑的观点:难道一次插入排序的时间复杂度会高于多次插入排序时间复杂度?

      通过切分方案,经过子数列的微排序(因子数列数字不多,其移动交换量也不会很大),最后一次插入排序的移动次数可以达到最小,只要增量选择合适,时间复杂度可以控制 在 O(n) 到 O(<sup>2</sup>) 之间。完全是有可能优于单纯的使用一次插入排序。

      3. 归并排序

      归并排序算法也是基于分治思想。和希尔排序一样,需要对原始数列进行切分,但是切分的方案不一样。

      相比较希尔排序,归并排序的分解子问题,求解子问题,合并子问题的过程分界线非常清晰。可以说,归并排序更能完美诠释什么是分治思想。

      3.1 分解子问题

      归并排序算法的分解过程采用二分方案。

      把原始数列一分为二。

      然后在已经切分后的子数列上又进行二分。

      如此反复,直到子数列不能再分为止。

      如下图所示:

      如下代码,使用递归算法对原数列进行切分,通过输出结果观察切分过程:

      # 切分原数列
      def split_nums(nums):
          print(nums)
          if len(nums) > 1:
              # 切分线,中间位置
              sp_line = len(nums) // 2
              split_nums(nums[0:sp_line])
              split_nums(nums[sp_line:])
      
      nums = [3, 9, 8, 1, 6, 5, 7]
      split_nums(nums)
      

      输出结果:和上面演示图的结论一样。

      [3, 9, 8, 1, 6, 5, 7]
      [3, 9, 8]
      [3]
      [9, 8]
      [9]
      [8]
      [1, 6, 5, 7]
      [1, 6]
      [1]
      [6]
      [5, 7]
      [5]
      [7]

      3.2 求解子问题

      切分后,对每相邻 2 个子数列进行合并。当对相邻 2 个数列进行合并时,不是简单合并,需要保证合并后的数字是排序的。如下图所示:

      3.3 合并排序

      如何实现 2 个数字合并后数字有序?

      使用子数列中首数字比较算法进行合并排序。如下图演示了如何合并 nums01=[1,3,8,9]、nums02=[5,6,7] 2 个子数列。

      子数列必须是有序的!!

      数字 1 和 数字 5 比较,5 大于 1 ,数字 1 先位于合并数列中。

      数字 3 与数字 5 比较,数字 3 先进入合并数列中。

      数字 8 和数字 5 比较,数字 5 进入合并数列中。

      从头至尾,进行首数字大小比较,最后,可以保证合并后的数列是有序的。

      编写一个合并排序代码:

      如果仅仅是合并 2 个有序数列,本文提供 2 个方案:

      不增加额外的存储空间:把最终合并排序好的数字全部存储到其中的一个数列中。

      def merge_sort(nums01, nums02):
          # 为 2 个数列创建 2 个指针
          idx_01 = 0
          idx_02 = 0
          while idx_01 < len(nums01) and idx_02 < len(nums02):
              if nums01[idx_01] > nums02[idx_02]:
                  # 这里不额外增加存储空间,如果数列 2 中的值大于数字 1 的插入到数列 1 中
                  nums01.insert(idx_01, nums02[idx_02])
                  idx_02 += 1
              # 数列 1 的指针向右移动    
              idx_01 += 1
          # 检查 nums02 中的数字是否已经全部合并到 nums01 中
          while idx_02 < len(nums02):
              nums01.append(nums02[idx_02])
              idx_02 += 1
      
      nums01 = [1, 2, 8, 9]
      nums02 = [5, 6, 7, 12, 15]
      merge_sort(nums01, nums02)
      # 合并后的数字都存储到了第一个数列中
      print(nums01)
      '''
      输出结果:
      [1,2,5,6,7,8,9,12,15]
      '''
      

      增加一个空数列,用来保存最终合并的数字。

      # 使用附加数列
      nums=[]
      def merge_sort(nums01, nums02):
          # 为 2 个数列创建 2 个指针
          idx_01 = 0
          idx_02 = 0
          k=0
          while idx_01 < len(nums01) and idx_02 < len(nums02):
              if nums01[idx_01] > nums02[idx_02]:
                  nums.append(nums02[idx_02])
                  idx_02 += 1
              else:
                  nums.append(nums01[idx_01])
                  idx_01 += 1
              k+=1
          # 检查是否全部合并
          while idx_02 < len(nums02):
              nums.append(nums02[idx_02])
              idx_02 += 1
          while idx_01 < len(nums01):
              nums.append(nums01[idx_01])
              idx_01 += 1
      
      nums01 = [1, 2, 8, 9]
      nums02 = [5, 6, 7, 12, 15]
      merge_sort(nums01, nums02)
      print(nums)
      

      前面是分步讲解切分和合并逻辑,现在把切分和合并逻辑合二为一,就完成了归并算法的实现:

      def merge_sort(nums):
          if len(nums) > 1:
              # 切分线,中间位置
              sp_line = len(nums) // 2
              nums01 = nums[:sp_line]
              nums02 = nums[sp_line:]
              merge_sort(nums01)
              merge_sort(nums02)
      
              # 为 2 个数列创建 2 个指针
              idx_01 = 0
              idx_02 = 0
              k = 0
              while idx_01 < len(nums01) and idx_02 < len(nums02):
                  if nums01[idx_01] > nums02[idx_02]:
                      # 合并后的数字要保存到原数列中
                      nums[k] = nums02[idx_02]
                      idx_02 += 1
                  else:
                      nums[k] = nums01[idx_01]
                      idx_01 += 1
                  k += 1
              # 检查是否全部合并
              while idx_02 < len(nums02):
                  nums[k] = nums02[idx_02]
                  idx_02 += 1
                  k += 1
              while idx_01 < len(nums01):
                  nums[k] = nums01[idx_01]
                  idx_01 += 1
                  k += 1
      
      nums = [3, 9, 8, 1, 6, 5, 7]
      merge_sort(nums)
      print(nums)
      

      个人觉得,归并算法对于理解分治思想有大的帮助。

      从归并算法上可以完整的体现分治理念的哲学之美。

      4. 基数排序

      基数排序(radix sort)属于“分配式排序”(distribution sort),又称“桶子法”(bucket sort)或 bin sort。

      基数排序没有使用分治理念,放在本文一起讲解,是因为基数排序有一个对数字自身切分逻辑。

      基数排序的最基本思想:

      如对原始数列 nums = [3, 9, 8, 1, 6, 5, 7] 中的数字使用基数排序。

      先提供一个长度为 10 的新空数列(本文也称为排序数列)。

      为什么新空数列的长度要设置为 10?等排序完毕,相信大家就能找到答案。

      把原数列中的数字转存到新空数列中,转存方案:

      nums 中的数字 3 存储在新数列索引号为 3 的位置。

      nums 中的数字 9 存储在新数列索引号为 9 的位置。

      nums 中的数字 8 存储在新数列索引号为 8 的位置。

      ……

      从上图可知,原数列中的数字所转存到排序数列中的位置,是数字所代表的索引号所指的位置。显然,经过转存后,新数列就是一个排好序的数列。

      新空数列的长度定义为多大由原始数列中数字的最大值来决定。

      编码实现:

      # 原数列
      nums = [3, 9, 8, 1, 6, 5, 7]
      # 找到数列中的最大值
      sort_nums=[0]*(max(nums)+1)
      for i in nums:
          sort_nums[i]=i
      
      print([i for i in sort_nums if i!=0])
      '''
      输出结果:
      [1,3,5,6,7,8,9]
      '''
      

      使用上述方案创建新空数据,如果数字之间的间隔较大时,新数列的空间浪费就非常大。

      如对 nums=[1,98,51,2,32,4,99,13,45] 使用上述方案排序,新空数列的长度要达到 99 ,真正需要保存的数字只有 7 个,如此空间浪费几乎是令人恐怖的。

      所以,有必要使用改良方案。如果在需要排序的数字中出现了 2 位以上的数字,则使用如下法则:

      先根据每一个数字个位上的数字进行存储。个位数是 1 存储在位置为 1 的位置,是 9 就存储在位置是 9 的位置。如下图:

      可看到有可能在同一个位置保存多个数字。这也是基数排序也称为桶排序的原因。

      一个位置就是一个桶,可以存放多个具有相同性质的数字。如上图:个位上数字相同的数字就在一个桶中。

      • 把存放在排序数列中的数字按顺序重新拿出来,这时的数列顺序变成 nums=[1,51,2,32,13,4,45,8,99]
      • 把重组后数列中的数字按十位上的数字重新存入排序数列。

      可以看到,经过 2 轮转存后,原数列就已经排好序。

      这个道理是很好理解的:

      现实生活中,我们在比较 2 个数字 大小时,可以先从个位上的数字相比较,然后再对十位上的数字比较。

      基数排序,很有生活的味道!!

      编码实现基数排序:

      nums = [1, 98, 51, 2, 32, 4, 99, 13, 45]
      # 数列中的最大值
      m = max(nums)
      # 确定最大位数,用来确定需要转存多少次
      l = len(str(m))
      
      for i in range(l + 1):
          # 排序数列,也是桶
          sort_nums = [[] for _ in range(10)]
          for n in nums:
              # 分解数字个位上的数字
              g_s = (n // 10 ** i) % 10
              # 根据个位上的数字找到转存位置
              sub_nums = sort_nums[g_s]
              sub_nums.append(n)
          # 合并数据
          nums = []
          for l in sort_nums:
              nums.extend(l)
      print(nums)
      '''
      输出结果:
      [1, 2, 4, 13, 32, 45, 51, 98, 99]
      '''
      

      上述转存过程是由低位到高位,也称为 LSD ,也可以先高位后低位方案转存MSD

      5. 总结

      分治很有哲学味道,当你遇到困难,应该试着找到问题的薄弱点,然后一点点地突破。

      当遇到困难时,老师们总会这么劝解我们。

      分治其实和项目开发中的组件设计思想也具有同工异曲之处。

      在线客服
      服务热线

      服务热线

      4008888355

      微信咨询
      二维码
      返回顶部
      ×二维码

      截屏,微信识别二维码

      打开微信

      微信号已复制,请打开微信添加咨询详情!