歸并排序中對小數組采用插入排序

from at003_insertsort import insertSort
from math import log
__author__ = 'Xiong Neng'
def mergeSort(seq):
 mergeSortRange(seq, 0, len(seq) - 1, log(len(seq)) - 1)
def mergeOrderedSeq(seq, left, middle, right):
 """
 seq: 待排序序列
 left <= middle <= right
 子數組seq[left..middle]和seq[middle+1..right]都是排好序的
 該排序的時間復雜度為O(n)
 """
 tempSeq = []
 i = left
 j = middle + 1
 while i <= middle and j <= right:
 if seq[i] <= seq[j]:
 tempSeq.append(seq[i])
 i += 1
 else:
 tempSeq.append(seq[j])
 j += 1
 if i <= middle:
 tempSeq.extend(seq[i:middle + 1])
 else:
 tempSeq.extend(seq[j:right + 1])
 seq[left:right + 1] = tempSeq[:]
def mergeSortRange(seq, start, end, threshold):
 """
 歸并排序一個序列的子序列
 start: 子序列的start下標
 end: 子序列的end下標
 threshold: 待排序長度低于這個值，就采用插入排序
 """
 if end - start < threshold:
 tempSeq = seq[start: end + 1]
 insertSort(tempSeq)
 seq[start: end + 1] = tempSeq[:]
 elif start < end: # 如果start >= end就終止遞歸調用
 middle = (start + end) / 2
 mergeSortRange(seq, start, middle, threshold) # 排好左邊的一半
 mergeSortRange(seq, middle + 1, end, threshold) # 再排好右邊的一半
 mergeOrderedSeq(seq, start, middle, end) # 最后合并排序結果
if __name__ == '__main__':
 aa = [4, 2, 5, 1, 6, 3, 7, 9, 8]
 mergeSort(aa)
 print(aa)