# 合并两个排序数组,生成一个新的数组
def merge(list1, list2):
res = []
i, j = 0, 0
while i < len(list1) and j < len(list2):
if list1[i] <= list2[j]:
res.append(list1[i])
i += 1
else:
res.append(list2[j])
j += 1
# 两个数组长度不一致的时候
res += list1[i:]
res += list2[j:]
return res
# 合并两个排序数组,将nums2的元素插入到nums1中,使得nums1成为一个有序数组
# 先合并到nums1中,再合并
def merge2(nums1, nums2):
nums1 += nums2
nums1.sort()
return nums1
if __name__ == "__main__":
a = [1, 2, 3, 5, 8]
b = [4, 5, 6]
print (merge(a, b))
print (merge2(a, b))