2.合并排序数组

# 原点
# 杨岩林
# 开发时间：2022-01-15 14:14
'''   合并两个升序的整数数组A和B，形成一个新的数组，新数组也要有序
   输入A=[1],B=[1], 输出[1,1],返回合并后的数组。输入A=[1,2,3,4],B=[2,4,5,6],输出[1,2,2,3,4,4,5,6],
   返回合并所有元素后的数组
   '''
class Solution:
   # 参数 A ：有序整数数组A
   # 参数 B ： 有序整数数组B
   # 返回 ：一个新的有序整数数
   def mergeSortedArray(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__':
    A=[1,4]
    B=[1,2,3]
    D=[1,2,3,4]
    E=[2,4,5,6]
    solution=Solution()
    print("输入：",A,"",B)
    print('输出：',solution.mergeSortedArray(A,B))
    print("输入：",D,"",E)
    print("输出：",solution.mergeSortedArray(D,E))

代码截图：

运行结果：

 # 为了遇见更好的自己，加油！