合并排序数组(LintCode)

题目来源:LintCode

原题地址:http://www.lintcode.com/zh-cn/problem/merge-sorted-array-ii/

题目:

合并两个排序的整数数组A和B变成一个新的数组。

您在真实的面试中是否遇到过这个题?  
Yes
样例

给出A=[1,2,3,4]B=[2,4,5,6],返回 [1,2,2,3,4,4,5,6]

挑战

你能否优化你的算法,如果其中一个数组很大而另一个数组很小?


难度级别:
容易

思路分析:
       我采用的比较常规的思路,用的就是归并排序的方法。
首先开辟一个新的空间,其大小是两个数组的总和,来存放合并排序之后的数组。
然后就是归并的常规思路了。
     对于挑战的意思,我的理解是否是要想办法来降低所需的额外空间呢? 还是要更快的进行合并排序呢?
     如果是降低所需的额外空间的话,那么如果数组是一大一小,且大的数组是能够容纳所有的元素的话。
那么,我们可以从数组的末尾开始从遍历,进行归并操作。当较小的数组元素全部归并结束之后,成像就可以停止归并了。

实现代码:
class Solution
{
public:
    /**
    * @param A and B: sorted integer array A and B.
    * @return: A new sorted integer array
    */
    vector mergeSortedArray(vector &A, vector &B)
    {
        vector result(A.size()+B.size(), 0);
        int i = 0, j = 0;
        int k = 0;
        while (i < A.size() && j < B.size())
        {
            if (A[i] > B[j])
            {
                result[k++] = B[j++];
            } else
            {
                result[k++] = A[i++];
            }
        }
        while (i < A.size())
        {
            result[k++] = A[i++];
        }
        while (j < B.size())
        {
            result[k++] = B[j++];
        }
        return result;
    }
};



代码说明:
上述代码采用的常规的思路,因为题中并未保证原数组的大小。

你可能感兴趣的:(算法,LintCode)