排序算法(python)-归并排序

排序算法(python)-归并排序

归并排序的思想就是先递归分解数组，再合并数组。

将数组分解最小之后，然后合并两个有序数组，基本思路是比较两个数组的最前面的数，谁小就先取谁，取了后相应的指针就往后移一位。然后再比较，直至一个数组为空，最后把另一个数组的剩余部分复制过来即可。

时间复杂度

• 最优时间复杂度：O(nlogn)
• 最坏时间复杂度：O(nlogn)
• 稳定性：稳定

对于一个int数组，请编写一个归并排序算法，对数组元素排序。

给定一个int数组A及数组的大小n，请返回排序后的数组。

测试样例：

[1,2,3,5,2,3],6
[1,2,2,3,3,5]

-- coding:utf-8 --

class MergeSort:

  def mergeSort(self, A, n):
      # write code here

      if n <= 1:
          return A
      mid = n / 2
      # left 采用归并排序后形成的有序的新的列表
      left_li = self.mergeSort(A[:mid], len(A[:mid]))
      # right 采用归并排序后形成的有序的新的列表
      right_li = self.mergeSort(A[mid:], len(A[mid:]))

      # 将两个有序的子序列合并为一个新的整体
      # merge(left, right)

      l , r = 0,0
      result = []

      while l < len(left_li) and r < len(right_li):
          if left_li[l] <= right_li[r]:
              result.append(left_li[l])
              l += 1
          else:
              result.append(right_li[r])
              r += 1
      result += left_li[l:]
      result += right_li[r:]

      return result

def merge_sort(alist):
    if len(alist) <= 1:
        return alist
    # 二分分解
    num = len(alist)/2
    left = merge_sort(alist[:num])
    right = merge_sort(alist[num:])
    # 合并
    return merge(left,right)

def merge(left, right):
    '''合并操作，将两个有序数组left[]和right[]合并成一个大的有序数组'''
    #left与right的下标指针
    l, r = 0, 0
    result = []
    while l<len(left) and r<len(right):
        if left[l] < right[r]:
            result.append(left[l])
            l += 1
        else:
            result.append(right[r])
            r += 1
    result += left[l:]
    result += right[r:]
    return result

alist = [54,26,93,17,77,31,44,55,20]
sorted_alist = mergeSort(alist)
print(sorted_alist)