Python3將兩個有序數組合並為一個有序數組

[本文出自天外歸雲的博客園]

第一種思路，把兩個數組合為一個數組然后再排序，問題又回歸到冒泡和快排了，沒有用到兩個數組的有序性。（不好）

第二種思路，循環比較兩個有序數組頭位元素的大小，並把頭元素放到新數組中，從老數組中刪掉，直到其中一個數組長度為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:
        ret += a[i:]
    if len(b) > j:
        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))

這個基本就是最簡單的方法了。