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

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

给定两个排序后的数组 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]

方法1：暴力法，使用额外的数组进行存储，之后再进行排序

class Solution {
public:
    void merge(vector<int>& A, int m, vector<int>& B, int n) {
        vector<int>res;
        for (int i = 0; i<m; i++) {
            res.push_back(A[i]);
        }
        for (int i = 0; i<n; i++) {
            res.push_back(B[i]);
        }
        sort(res.begin(), res.end());
        A = res;
    }
};

方法2：使用set对数组元素存储，自动排序

class Solution {
public:
    multiset<int>mySet;

public:
    void merge(vector<int>& A, int m, vector<int>& B, int n) {
        for (int i = 0; i<m; i++) {
            mySet.insert(A[i]);
        }
        for (int i = 0; i<n; i++) {
            mySet.insert(B[i]);
        }
        A.clear();
        auto iter = mySet.begin();
        for (; iter != mySet.end(); iter++) {
            A.push_back(*iter);
        }
    }
};