合并两个有序数组python_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))
這個基本就是最簡單的方法了。
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持腳本之家。
總結
以上是生活随笔為你收集整理的合并两个有序数组python_Python3合并两个有序数组代码实例的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 最简单人工智能python_Python
- 下一篇: python import reques