之前一直没有练习,但记得思想。归并时间复杂度O(nlogn),并且稳定,有必要练习下。
思路:数组先做二分,直到只有一个元素时候返回(自然用到栈,并且空间复杂度是O(n),每层调用排序函数后会释放栈帧,所以不需要累加)。从递归树最深层开始排序,排序思想很简单,两个数组给两个指针,每次比较两个指针的元素的大小,小的进入新的数组,移动指针,最后记得带上剩下的数组元素。
可以参考图片
class Merge(object):
def sort(self, l, r):
j = 0;k = 0
temp = []
while j<len(l) and k<len(r):
if l[j] < r[k]: temp.append(l[j]); j += 1
else: temp.append(r[k]); k += 1
temp.extend(l[j:])
temp.extend(r[k:])
return temp
def sp(self, ls):
if len(ls) // 2 == 0: return ls
mid = len(ls) // 2
l = self.sp(ls[:mid])
r = self.sp(ls[mid:])
return self.sort(l, r)
ls = [2, 1, 3, 0, 4, 5, 5, 7, 8]
solu = Merge()
print(solu.sp(ls))