【算法百题之五十七】删除排序数组中的重复项

【算法百题之五十七】删除排序数组中的重复项

 

   大家好，我是Lampard~~

   很高兴又能和大家见面了，接下来准备系列更新的是算法题，一日一练，早日升仙！

   今天的问题是：删除排序数组中的重复项，要求时间复杂度是o(1)

示例：

给定数组 nums = [1,1,2], 

函数应该返回新的长度 2, 并且原数组 nums 的前两个元素被修改为 1, 2。 

 

给定 nums = [0,0,1,1,1,2,2,3,3,4],

函数应该返回新的长度 5, 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4。

思路：这道是比较简单，考察的点是对vector迭代器删除函数的调用，我们只需要一次循环，利用一个map记录已经出现的数，若在map中找不到当前循环的数nums【i】,则把这个数用map记录下来，并判断下一个数。若已经在map中被记录了，那么调用erase函数删除该元素即可

算法代码：

int removeDuplicates(vector<int>& nums) {
	map<int, bool> num2Index;
	int num = 0;
	for (vector<int>::iterator it = nums.begin(); it != nums.end();) {
		if (num2Index.count(*it) == 0) {
			num2Index[*it] = num;
			num++;
			it++;
		}
		else {
			it = nums.erase(it);
		}
	}

	return num;
}

这里有一个注意的地方：

调用erase（）函数后，vector后面的元素会向前移位，此时原先的迭代器已经失效。所以这样写是不对的

int removeDuplicates(vector<int>& nums) {
	map<int, bool> num2Index;
	int num = 0;
	for (vector<int>::iterator it = nums.begin(); it != nums.end(); it++，num++) {
		if (num2Index.count(*it) == 0) {
			num2Index[*it] = num;
		}
		else {
			nums.erase(it);
		}
	}

	return num;
}

测试结果：

 

OK，今天的博客就到这里，谢谢大家！！！