LintCode笔记（9）——合并排序数组 与vector的insert和erase总结笔记

合并两个排序的整数数组A和B变成一个新的数组。

您在真实的面试中是否遇到过这个题？ 

Yes

样例

给出A=[1,2,3,4]，B=[2,4,5,6]，返回 [1,2,2,3,4,4,5,6]

挑战 

你能否优化你的算法，如果其中一个数组很大而另一个数组很小？

这道题题目的原理很简单，即便是挑战的内容也很简单，就是遍历两个数组，比较两个数组中正在遍历的部分，哪个小就把哪个先放入结果数组中，直到结果数组的总长度为两个数组长度之和则合并完毕。挑战内容无非就是把A或B当作结果数组。

这道题我要总结的主要原因就是vector的insert和erase两个函数的用法。

1、vector的Insert用法

vector的insert用法主要由以下三种：

（1）直接插入一个元素

A.insert(A.begin(),5);//把5插入到数组的第1个元素
A.insert(A.being()+2,6);//把6插入到数组的第3个元素

这里需要注意的是，首先给定插入位置的是iterator类型的。

还有一点需要注意，这个函数是有返回值的，它返回的也是一个iterator类型的对象，返回的是插入的位置，即

	A.push_back(1);
	vector<int>::iterator Iter2;
	Iter2 = A.begin();
	Iter2 = A.insert(A.begin(), 4);//数组的第一个元素位置处插入4
	if (Iter2 == A.begin())
		cout << "Iter2 == A.being()" << endl;

最终输出结果为Iter2 == A.begin()

（2）重复插入多个元素

	A.push_back(1);
	vector<int>::iterator Iter2;
	Iter2 = A.begin();
	int numberToInsert = 5;
	int times = 4
	Iter2 = A.insert(A.begin(), times, numberToInsert);//数组的第一个元素位置处插入4个5
	for (int i = 0; i < A.size(); i++)
	{
		cout << A[i] << endl;
	}
	if (Iter2 == A.begin())
		cout << "Iter2 == A.begin()" << endl;

同样，插入后Iter2的值并没有改变。

（3）插入其他vector的一段

	vector<int> A;
	vector<int> B;
	B.push_back(2);
	B.push_back(3);
	B.push_back(4);
	A.push_back(1);
	vector<int>::iterator Iter2;
	Iter2 = A.begin();
	Iter2 = A.insert(A.begin(), B.begin(),B.begin()+2);
	for (int i = 0; i < A.size(); i++)
	{
		cout << A[i] << endl;
	}
	if (Iter2 == A.begin())
		cout << "Iter2 == A.begin()" << endl;

这里需要注意的是：

A.insert(A.begin(), B.begin(),B.begin()+2);

这里只插入B的第1个和第2个元素，而第3个元素并不插入，即对于插入的B的区间是半开半闭的即 [ )

同样Iter2插入前后的值不发生改变。

2、vector的erase函数

vector中用于删除元素的方法主要由三个<algorithm>头文件的remove()和vector自带的pop_back()，erase();

pop_back()非常简单，就是删去数组中的最后一个元素；

remove()方法就有些复杂了，他和erase()的最大不同就是它删除元素后，整个vector 的大小不会发生改变，这里的原因是，如果他删除了某个元素，那么数组中在该元素之前的元素不动，该元素之后的元素分别向前移动一个位置，覆盖前面一个元素的是，但是原数组中最后一个位置并不删除，保留下来。

如下代码所示：

int main()
{
	vector<int> A;
	for (int i = 0; i < 10; i++)
		A.push_back(i);
	cout << "原数组为：" << endl;
	for (int i = 0; i < A.size(); i++)
	{
		cout << A[i] << endl;
	}
	remove(A.begin(), A.end(), 3);//去除A中从begin到end中值为3的元素
	cout << "利用remove删除后的数组：" << endl;
	for (int i = 0; i < A.size(); i++)
	{
		cout << A[i] << endl;
	}
}

得到的结果为

erase的用法就比较简单，使用iterator指定删除某个元素，但是需要注意的是删除后该iterator成为野指针，不能直接继续用，需要重新赋值。

vector<int> A;
	for (int i = 0; i < 10; i++)
		A.push_back(i);
	cout << "原数组为：" << endl;
	for (int i = 0; i < A.size(); i++)
	{
		cout << A[i] << endl;
	}
	vector<int>::iterator iter;
	//这样做是不行的，会使iter成为野指针
	//for (iter = A.begin(); iter != A.end(); iter++)
	//	A.erase(iter)

	//如下赋值后iter成为原删除元素下一个元素的指针但是这里有一个问题
	//不能删除连续的2个3，因为后面for循环有iter++
	//for (iter = A.begin(); iter != A.end(); iter++)
	//	if (*iter == 3)
	//		iter = A.erase(iter);

	//这样做是最好的
	for (iter = A.begin(); iter != A.end();)
	{
		if (*iter == 3)
			iter = A.erase(iter);
		else
			iter++;
	}

	cout << "使用erase删除后数组为：" << endl;
	for (int i = 0; i < A.size(); i++)
	{
		cout << A[i] << endl;
	}

最后结果A为0,1,2,4,5,6,7,8,9

最后有一个remove和erase结合的例子，简单方便

int main()
{
	vector<int> A;

	for (int i = 0; i < 10; i++)
		A.push_back(i);
	A.insert(A.begin() + 5, 2, 3);//在A的第6个元素出插入2个3
	cout << "原数组为：" << endl;
	for (int i = 0; i < A.size(); i++)
	{
		cout << A[i] << endl;
	}

	//先使用remove去除重复的3，再使用erase去除数组中的末尾部分
	//使用erase需要包含<algorigthm>头文件
	A.erase(remove(A.begin(), A.end(), 3), A.end());

	cout << "使用erase和remove删除后数组为：" << endl;
	for (int i = 0; i < A.size(); i++)
	{
		cout << A[i] << endl;
	}
}

最后结果为：

好了说了这么多，合并排序数组这道题的代码如下：

class Solution {
public:
    /**
     * @param A and B: sorted integer array A and B.
     * @return: A new sorted integer array
     */
vector<int> mergeSortedArray(vector<int> &A, vector<int> &B) 
{
	// write your code here
	vector<int>::iterator IterA = A.begin();
	vector<int>::iterator IterB = B.begin();
	int size = A.size() + B.size();
	while (A.size() < size)
	{
		if (IterA == A.end())
		{
			
			while (B.size() != 0)
			{
				IterA = A.insert(IterA, *IterB);
				B.erase(IterB);
				IterB = B.begin();
				IterA ++;
			}
			break;
		}
		if (*IterA < *IterB)
			IterA++;
		else
		{
			IterA = A.insert(IterA, *IterB);
			B.erase(IterB);
			IterB = B.begin();
			IterA++;
		}
	}
	return A;
}
};