码迷,mamicode.com
首页 > 编程语言 > 详细

Python版将两个有序数组合并为一个有序数组

时间:2018-05-01 20:20:16      阅读:497      评论:0      收藏:0      [点我收藏+]

标签:return   替代   代码   排序算法   方法   比较   排序   也会   有序性   

第一种思路,把两个数组合为一个数组然后再排序,问题又回归到冒泡和快排了,没有用到两个数组的有序性。(不好)

第二种思路,循环比较两个有序数组头位元素的大小,并把头元素放到新数组中,从老数组中删掉,直到其中一个数组长度为0。然后再把不为空的老数组中剩下的部分加到新数组的结尾。(好)

第二种思路的排序算法与测试代码如下:

def merge_sort(a, b):
    ret = []
    while len(a)>0 and len(b)>0:
        if a[0] <= b[0]:
            ret.append(a[0])
            a.remove(a[0])    
        if a[0] >= b[0]:
            ret.append(b[0])
            b.remove(b[0])
    if len(a) == 0:
        ret += b
    if len(b) == 0:
        ret += a
    return ret


if __name__ == __main__:
    a = [1,3,4,6,7,78,97,190]
    b = [2,5,6,8,10,12,14,16,18]
    print(merge_sort(a, b))

反思了一下上面的过程,不应该用remove方法,因为仔细想一下remove方法里面也会涉及遍历,不算最简单。

改进一下,改用索引元素比较法替代头位元素比较法:

def merge_sort(a, b):
    ret = []
    i = j = 0
    while len(a) > i + 1 and len(b) > j + 1:
        if a[i] <= b[j]:
            ret.append(a[i])
            i += 1
        else:
            ret.append(b[j])
            j += 1
    if len(a) > i + 1:
        ret += a[i:]
    if len(b) > j + 1:
        ret += b[j:]
    return ret

if __name__ == __main__:
    a = [1,3,4,6,7,78,97,190]
    b = [2,5,6,8,10,12,14,16,18]
    print(merge_sort(a, b))

这个基本就是最简单的方法了。

 

Python版将两个有序数组合并为一个有序数组

标签:return   替代   代码   排序算法   方法   比较   排序   也会   有序性   

原文地址:https://www.cnblogs.com/LanTianYou/p/8976671.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!