题目来源: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<int> mergeSortedArray(vector<int> &A, vector<int> &B)
{
vector<int> 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;
}
};
代码说明:
上述代码采用的常规的思路,因为题中并未保证原数组的大小。
上述代码采用的常规的思路,因为题中并未保证原数组的大小。