归并排序算法的python实现

归并算法是一种常见的算法，其原理如下：

假如有2组数据list1,list2，在每组中数据都是有序的，假定都是从小到大排序，那么构建新的list

然后比较list1和list2的第一个元素的，依次把较小的写入list中。比如

list1=[3,5,7,8] list2=[2,4,9,11]

那么，就会按照2,3,4,5,7,8,9,11的顺序写入list，定义这个过程定义为函数merge

归并算法本质就是把列表一直分拆下去，直至只有一个元素在列表中，1个元素自然不需要排序，

然后依次merge这些被分拆的size为1的列表，合成整个结果列表

因此，先定义merger函数

def merge(ps1,ps2):
    ans=[]
    i,j=0,0
    while i<len(ps1) and j<len(ps2):
        if ps1[i]<=ps2[j]:
            ans.append(ps1[i])
            i+=1
        else:
            ans.append(ps2[j])
            j+=1
    ans+=ps1[i:]
    ans+=ps2[j:]
    return ans

剩下的工作，就是一个递归过程

def mergeSort(ps):
    if len(ps)<=1:
        return ps
    temp=len(ps)//2
    ps1=ps[:temp]
    ps2=ps[temp:]
    return merge(mergeSort(ps1),mergeSort(ps2))

当然，利用python的reduce函数，还有另外一种mergeSort的方法，就是先把列表分拆为以每个元素为一个列表的列表

即假定列表为[3,4,2,1,5]，先变为[[3],[4],[2],[1],[5]]

然后就可以用reduce方法，具体实现如代码所示

def mergeSort(ps):
    def split():
        ans=[]
        for p in ps:
            tem=[]
            tem.append(p)
            ans.append(tem)
        return ans
    from functools import reduce #reduce在3.4版本中在functools模块中，使用前要先import
    return reduce(merge,split())

这样做，好像违背了python的只有一种方法的精神

其实，还可以尝试修改merge，使得merge变得强大，不仅可以merge列表，还可以merge数字，以及merge数字和列表

最简单的修改方式就是，假定输入的一个参数不是列表，则使其变为

#mergesort coding by Aris
def merge(ps1,ps2):
    def transList(p):
        ans=[]
        ans.append(p)
        return ans
    def noTrans(p):
        return p
    funcDict={True:noTrans,False:transList} #利用字典做switch选择
    ps1=funcDict[type(ps1)==list](ps1)
    ps2=funcDict[type(ps2)==list](ps2)
    ans=[]
    i,j=0,0
    while i<len(ps1) and j<len(ps2):
        if ps1[i]<=ps2[j]:
            ans.append(ps1[i])
            i+=1
        else:
            ans.append(ps2[j])
            j+=1
    ans+=ps1[i:]
    ans+=ps2[j:]
    return ans

其中，python中没有switch选择语句，可以像上面代码中那样，使用字典的方式实现switch

这样修改了merge函数后，就可以用下面的语句，实现归并排序了

def mergeSort(ps):
    from functools import reduce
    return reduce(merge,ps)

其实就是一句 reduce