例2合并排序数组

"""
设计：Python程序设计
作者：初学者
日期：2022年 03月 21日
"""


# 例2合并排序数组
# 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.代码实现
class Solution:
    """
    参数A：有序整数数组 A
    参数B：有序整数数组 B
    返回：一个新的有序整数数组
    """

    def array(self, A, B):
        i, j = 0, 0
        C = []
        while i < len(A) and j < len(B):
            if A[i] < B[j]:
                C.append(A[i])
                i += 1
            else:
                C.append(B[j])
                j += 1
        while i < len(A):
            C.append(A[i])
            i += 1
        while j < len(B):
            C.append(B[j])
            j += 1
        return C


# 主函数
if __name__ == '__main__':
    s = Solution()
    A = [2, 3]
    B = [2, 3, 5]
    E = [4, 5, 6, 7]
    F = [5, 6, 7, 8, 9]
    print('输入：', A, " ", B)
    print("输出：", s.array(A, B))
    print('输入：', E, " ", F)
    print("输出：", s.array(E, F))