LeetCode - Merge Sorted Array
题意
把两个排好序的数组合并到第一个里。
思路
从后面开始合并,这样不用移动原来数组中的元素。
代码
class Solution {
public:
void merge(int A[], int m, int B[], int n)
{
int i = m - 1, j = n - 1, pos = m + n - 1;
while (1)
{
if (j < 0)
{
while (i >= 0) A[pos--] = A[i--];
break;
}
if (i < 0)
{
while (j >= 0) A[pos--] = B[j--];
break;
}
if (A[i] < B[j]) A[pos--] = B[j--];
else A[pos--] = A[i--];
if (pos < 0) break;
}
}
};