Python3合併兩個有序陣列程式碼例項
阿新 • • 發佈:2020-08-13
第一種思路,把兩個數組合為一個數組然後再排序,問題又迴歸到冒泡和快排了,沒有用到兩個陣列的有序性。(不好)
第二種思路,迴圈比較兩個有序陣列頭位元素的大小,並把頭元素放到新陣列中,從老陣列中刪掉,直到其中一個數組長度為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,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,b))
這個基本就是最簡單的方法了。
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。