首页 > 代码库 > [LeetCode] Merge Sorted Array

[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];    }}

 

[LeetCode] Merge Sorted Array