合并两个有序的数组

题目链接

Solution

因为AB都是有序的,所以定义两个指针,从前往后依次比较当前元素,小的加入到新的队列中,然后指针右移。
类似归并排序中的归并操作。

class Solution {
public:
    void merge(int A[], int m, int B[], int n) {
        vector<int> C;
        int i = 0, j = 0;
        while (i < m && j < n) {
            if (A[i] < B[j]) C.push_back(A[i++]);
            else C.push_back(B[j++]);
        }
        while (i < m) C.push_back(A[i++]);
        while (j < n) C.push_back(B[j++]);
        for (int i = 0; i < n + m; ++i) A[i] = C[i];
    }
};