leetcode 80:Remove Duplicates from Sorted Array II

题目：

Follow up for "Remove Duplicates":
What if duplicates are allowed at most twice?

For example,
Given sorted array nums = [1,1,1,2,2,3],

Your function should return length = 5, with the first five elements of nums being 1, 1, 2, 2 and 3. It doesn't matter what you leave beyond the new length.

思路：

因为已经排好序了，我们只需看每个元素相同的个数，超过2个我们就删除多余的，然后将后面的元素往前移动，移动操作直接借用STL的move来完成。

但是从后面开始检测，更节省时间。

实现如下：

class Solution {
public:
	int removeDuplicates(vector<int>& nums) {
		int size = nums.size();
		if (size < 3) return size;
		int count = 1,flag=0;
		vector<int>::iterator last = nums.end();
		for (vector<int>::iterator i = nums.begin() + 1; i < last; ++i)
		{
			while (i != last && *i == *(i - 1))
			{
				++count;
				++i;
				flag = 1;
			}
			if (count >= 3)
			{
				last -= count - 2;
				move(i, nums.end(), i - count + 2);
				i -= (count - 2);
				count = 1;
			}
			else count = 1;
			if (flag == 1)
			{
				flag = 0;
				--i;
			}
		}
		return (last - nums.begin());
	}
};

下面提供更快、更简洁的方法，不需要移动数据。

记录元素个数，当第i个元素与前两个不相同时，就将该元素存到相应位置。

时间复杂度：O(n)

class Solution {
public:
	int removeDuplicates(vector<int>& nums) {
		int size = nums.size();
		if (size < 3) return size;
		int rear = 1;
		for (int i = 2; i < size; ++i)
			if (!(nums[i] == nums[rear] && nums[i] == nums[rear - 1])) nums[++rear] = nums[i];
		return rear + 1;
	}
};