Merge Sorted Array
Given two sorted integer arrays A and B, merge B into A as one sorted array.
Note:
You may assume that A has enough space to hold additional elements from B. The number of elements initialized in A and B are m and n respectively.
这是一个逆思维,不是从前往后填写,而是从后往前填写,这样就可以避免不断的移动数据,只要一次遍历就可以了。
而且不需要使用额外的O(n+m)空间。这里应该是使用了O(m)空间吧,m代表B的长度。
严慧敏的数据结构书也有类似算法,需要O(n+m)的空间复杂度。
编程之美也有类似的题目,也差不多是这里这样的解法。
void merge2(int A[], int m, int B[], int n)
{
for (int i = m+n-1; i >= 0; i--)
{
if (m>0 && n>0)
{
if (A[m-1] < B[n-1]) A[i] = B[--n];
else A[i] = A[--m];
}
else if (m<=0 && n>0)
{
A[i] = B[--n];
}
else if (m>0 && n<=0)
{
A[i] = A[--m];
}
}
}
参考一下leetcode上的程序,发现,原来可以简洁成这样,极品:
void merge(int A[], int m, int B[], int n)
{
while (n>0)
{
if (m<=0 || A[m-1] < B[n-1]) A[n+m-1] = B[--n];
else A[n+m-1] = A[--m];
}
}
//2014-2-14 update
void merge(int A[], int m, int B[], int n)
{
int len = m + n -1;
for (int i = m - 1, j = n - 1; j >= 0 ; )
{
if (i>=0 && A[i] > B[j]) A[len--] = A[i--];
else A[len--] = B[j--];
}
}