python两个数组合并排序_2、合并排序数组

1、问题描述

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

2、问题示例

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

输入A=[2, 4, 6, 7]，B=[1, 3, 5, 9]，输出[1, 2, 3, 4, 5, 6, 7, 9]，返回合并所有元素后的数组。

3、代码实现

1)分别给两数组设置一个位置标识，如数组A的位置标识是i，数组B的位置标识为j，一开始 i、j 为0，表示两数组的第一个数据的位置；

2)新建一个空数组C，用来存放最终的结果数据；

3)使用while进行循环判断，当 i 小于数组A的长度并且 j 小于数组B的长度的前提下，

如果数组A的位于 i 位置的数据小于数组B的位于 j 位置的数据，就把数组A中 i 位置的数据添加到数组C中，同时 i 加一，表示数组A位于 i 位置的数据已处理，下次处理当前位置的下一位置(即i+1位置)的数据；

反之，如果数组B的位于 j 位置的数据小于数组A的位于 i 位置的数据，就把数组B中 j 位置的数据添加到数组C中，同时 j 加一，表示数组B位于 j 位置的数据已处理，下次处理当前位置的下一位置(即j+1位置)的数据；

直到某一数组的元素都已被处理完，即 i 大于或等于 数组A的长度，或者 j 大于或等于数组B的长度，则循环终止。

4)对两数组进行判断，看还有哪个数组还有数据没处理完，即存在 i 小于数组A的长度或者 j 小于数组B的长度，则把该数组剩余的数组直接都添加的数组C中即可。

class Solution:

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__":

arrA = [2, 4, 6, 7]

arrB = [1, 3, 5, 9]

solution = Solution()

arrC = solution.mergeSortedArray(arrA, arrB)

print("A:", arrA)

print("B:", arrB)

print("C:", arrC)