例002-合并排序数组

例002 合并排序数组

1.问题描述

合并两个升序的整数数组A和B，形成一个新的数组，新数组也要有序。

2.问题示例

输入： A= [1] , B = [1]
输出： [1,1]
返回合并后的数组

输入： A = [1,2,3,4], B = [2,4,5,6]
输出： [1,2,2,3,4,4,5,6]
返回合并后的数组

3.代码实现

def mergeSortedArray(A,B):
    i = 0
    j = 0

    out_array = []

    # 比较两个数组的数值，按从小到大的顺序排列输出，不能把两个数组的数值全部遍历完
    while i < len(A) and j < len(B):
        if A[i] < B[j]:
            out_array.append(A[i])
            i += 1
        else:
            out_array.append(B[j])
            j += 1
    # 如果A数组的值比较大 ，
    while i < len(A):
        out_array.append(A[i])
        i += 1
    # 如果 B数组的值比较大
    while j < len(B):
        out_array.append(B[j])
        j += 1

    return out_array

A = [1,4]
B = [1,2,3]
C = [1,2,3,4]
D = [2,4,5,6]



print('输入：',A,'  ',B)
print('输出：',mergeSortedArray(A, B))

print('输入：',C,'  ',D)
print('输出：',mergeSortedArray(C, D))



输入： [1, 4]    [1, 2, 3]
输出： [1, 1, 2, 3, 4]
输入： [1, 2, 3, 4]    [2, 4, 5, 6]
输出： [1, 2, 2, 3, 4, 4, 5, 6]