LintCode 合并排序数组 II

合并排序数组 II

合并两个排序的整数数组A和B变成一个新的数组。
样例
给出A = [1, 2, 3, empty, empty] B = [4,5]
合并之后A将变成[1,2,3,4,5]
注意
你可以假设A具有足够的空间（A数组的大小大于或等于m+n）去添加B中的元素。

方法一：开辟一个临时数组tmp来存放A中前面m个数字，那么就变成了归并tmp和B，把结果存放在A

class Solution {
public:
    /**
     * @param A: sorted integer array A which has m elements, 
     *           but size of A is m+n
     * @param B: sorted integer array B which has n elements
     * @return: void
     */
    void mergeSortedArray(int A[], int m, int B[], int n) {
        // write your code here
        vector<int> tmp(m, 0);
        int i =0, j = 0, k = 0;
        for (int i = 0; i < m; ++i)
            tmp[i] = A[i];
        while(i < m && j < n) {
            if (tmp[i] < B[j])
                A[k++] = tmp[i++];
            else 
                A[k++] = B[j++];
        } 
        while(i < m)
            A[k++] = tmp[i++];
        while(j < n)
            A[k++] = B[j++];
    }
};

方法二：如果不能开辟额外的空间， 那么方法一是肯定不行的，那么我们可以把B中的元素插入到A中m个元素的后面，然后在对A进行快速排序。
注意：该方法存在超时问题

class Solution {
public:
    /**
     * @param A: sorted integer array A which has m elements, 
     *           but size of A is m+n
     * @param B: sorted integer array B which has n elements
     * @return: void
     */
    void mergeSortedArray(int A[], int m, int B[], int n) {
        // write your code here
        int i =m, j = 0;
        while(j < n)
            A[i++] = B[j++];
        quickSort(A, 0, m+n-1);
    }
private :
    void quickSort(int A[], int p, int r)
    {
        if (p < r) {
            int q = partition(A,p, r);
            quickSort(A, p, q-1);
            quickSort(A, q+1, r);
        }
    }

    int partition(int A[], int p, int r)
    {
        int x = A[r];
        int i = p-1;
        for ( int j = p; j <= r-1; ++j) {
            if (A[j] <= x) {
                ++i;
                swap(A[i], A[j]);
            } 
        }
        swap(A[i+1], A[r]);
        return i + 1;
    }

    inline void swap(int &a, int &b)
    {
        int c = a;
        a = b;
        b = c;
    }
};