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、如果数组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)