[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 = 0, j = 0;
    for (int k = 0; 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 = 0; k < m+n; k++) {
        A[k] = aux[k];
    }
}
原文地址:https://www.cnblogs.com/agentgamer/p/4085637.html