51 合并排序数组（Merge Two Sorted Arrays）

1 题目

题目：合并排序数组（Merge Two Sorted Arrays）
描述：将按升序排序的整数数组A和B合并，新数组也需有序。

lintcode题号——6，难度——easy

样例1：

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

样例2：

输入：
A = [1,2,3,4]
B = [2,4,5,6]
输出：
[1,2,2,3,4,4,5,6]

2 解决方案

2.1 思路

  将两个已排序的数组合并到新数组中，从两个数组头部开始，对比大小，将较小的放入新数组，以此类推直到其中一个为空，再将剩下不空的数组中的元素一次放入新数组即可完成排序。

2.3 时间复杂度

  若两个序列的长度分别为m和n，时间复杂度为O(m + n)。

2.4 空间复杂度

  空间复杂度为O(1)。

3 源码

细节：

1. 与有序链表的合并方式的区别在于当其中一个序列为空时，需要通过循环将不空的序列剩余元素放入新数组。

C++版本：

/**
* @param A: sorted integer array A
* @param B: sorted integer array B
* @return: A new sorted integer array
*/
vector<int> mergeSortedArray(vector<int> &A, vector<int> &B) {
    // write your code here
    vector<int> result;
    int i = 0;
    int j = 0;
    while (i < A.size() && j < B.size())
    {
        if (A.at(i) < B.at(j))
        {
            result.push_back(A.at(i++));
        }
        else
        {
            result.push_back(B.at(j++));
        }
    }

    while (i < A.size())
    {
        result.push_back(A.at(i++));
    }
    while (j < B.size())
    {
        result.push_back(B.at(j++));
    }

    return result;
}