![914138ca63cdacc211511182d0a06814.png](https://img-blog.csdnimg.cn/img_convert/914138ca63cdacc211511182d0a06814.png)
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、如果数组A中位于 i 位置的数据小于数组 B 中位于 j 位置的数据,就把数组 A 中 i 位置的数据添加到数组 C 中,同时 i 加 1,
表示数组 A 位于 i 位置的数据已处理,下次处理当前位置的下一位置(即 i+1 位置)的数据;
b、反之,如果数组 B 中位于 j 位置的数据小于数组 A 中位于 i 位置的数据,就把数组B中 j 位置的数据添加到数组C中,同时 j 加 1,
表示数组 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)