【python编程300例】第2例—合并排序数组

例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:
    # 类里面定义一个方法
    def fangfa(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


A = [1, 2, 3, 4]
B = [2, 3, 4, 5, 7, 8, 9]
solution = Solution()
print("输出：", solution.fangfa(A, B))