程序员面试金典 - 面试题 10.01. 合并排序的数组

1. 题目

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sorted-merge-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

• 依次比较大小插入A中即可

class Solution {
public:
    void merge(vector<int>& A, int m, vector<int>& B, int n) {
    	int i = 0, j = 0, count = m+n;
    	while(m && n)
    	{
    		if(A[i] <= B[j])
    			i++,m--;
    		else
    		{
    			A.insert(A.begin()+i, B[j++]);
    			n--;
    			i++;
    		}
    	}
    	while(n--)
    		A[i++] = B[j++];
    	while(A.size() > count)
    		A.pop_back();
    }
};