合并两个排序数组，python实现

# 合并两个排序数组，生成一个新的数组
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))