[LeetCode] Merge Sorted Array

时间:2023-03-09 06:26:55
[LeetCode] 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 (size that is greater or equal to m + n) to hold additional elements from B. The number of elements initialized in A and B are m and nrespectively.

把两个有序数组合并成一个有序数组,这是merge sort的基础步骤,因为在algs4里面有训练过所以一下就写出来了,不过第一次遇到的时候我是没能写出来,原因是有tricky的corner case,就是扫描两个数组的时候有可能其中一个数组先结束了,这时候就要从另一个数组里面拿数据。 这里的办法是先设定总的迭代次数,就是m+n,反正不会超过这个量,然后在里面分别判断两个数组有无越界,有则从另一个数组拿数据,否则比较两数据大小,把小的取走。

void merge(int A[], int m, int B[], int n) {
int aux[m+n];
int i = , j = ;
for (int k = ; k < m+n; k++) {
if (i >= m) { aux[k] = B[j++]; }
else if (j >= n) { aux[k] = A[i++]; }
else if (A[i] < B[j]) {
aux[k] = A[i++];
} else {
aux[k] = B[j++];
}
} for (int k = ; k < m+n; k++) {
A[k] = aux[k];
}
}