[LeetCode] Merge Sorted Array

标签：style   blog   io   color   ar   sp   for   数据   div   

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

标签：style   blog   io   color   ar   sp   for   数据   div   

原文地址：http://www.cnblogs.com/agentgamer/p/4085637.html