面试题 10.01. 合并排序的数组

Tags: Easy Two Pointers Array

Links: https://leetcode-cn.com/problems/sorted-merge-lcci/

---

给定两个排序后的数组 A 和 B，其中 A 的末端有足够的缓冲空间容纳 B。 编写一个方法，将 B 合并入 A 并排序。

初始化 A 和 B 的元素数量分别为 m 和 n。

示例:

输入:
A = [1,2,3,0,0,0], m = 3
B = [2,5,6],       n = 3

输出: [1,2,2,3,5,6]

说明:

• A.length == n + m

---

class Solution {
public:
    void merge(vector<int>& A, int m, vector<int>& B, int n) {
        std::ios_base::sync_with_stdio(false);
        cin.tie(NULL);
        cout.tie(NULL);

        int p1 = m - 1, p2 = n - 1;
        int pos = m + n - 1;
        while (p1 >= 0 && p2 >= 0) {
            if (A[p1] >= B[p2]) {
                A[pos] = A[p1]; --p1;
            }
            else {
                A[pos] = B[p2]; --p2;
            }
            --pos;
        }

        if (p2 >= 0) {
            for (int i = 0; i <= pos; ++i) {
                A[i] = B[i];
            }
        }
    }
};

比较末端的数字大小即可。