[LeetCode]Merge Sorted Array

Description:

 

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 arem and n respectively.

 

描述：

给出A和B两个已经排好序的整数数组（从小到大排序），把B整合入A中，让整合后的数组也是排好序的。另外，A有足够多的空间来存储数据（即A的长度大于等于m＋n，m和n为两个数组所包含数据的大小）。

 

解答：

归并排序的步骤之一。只用两个数组可以节省空间。至于节省时间的方法，就是我们要想的答案了。从尾部开始排序——哪一个数据大，就先放到尾部。其余的部分就和大家熟悉的归并排序一样了。但是这道题注意边界条件。

C++代码：

class Solution {
public:
    void merge(int A[], int m, int B[], int n)
    {
        if(m == 0) {
            for(int j = 0; j < n; j++) A[j] = B[j];
            return;
        }
        
        if(n == 0) {
            return;
        }
        
        int index = m + n - 1;
        int i = m - 1;
        int j = n - 1;
        
        while(i >= 0 && j >= 0) {
            if( A[i] > B[j] ) {
                A[index] = A[i];
                i--;
                index--;
            } else {
                A[index] = B[j];
                j--;
                index--;
            }
        }
        
        while(i >= 0) {
            A[index] = A[i];
            i--;
            index--;
        }
        
        while(j >= 0) {
            A[index] = B[j];
            j--;
            index--;
        }
    }
};