【leetcode】80. Remove Duplicates from Sorted Array II

Difficulty：Medium

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.

解题思路：

完成题目要求的取出有3个重复的数字，并且题目中所给出的序列都是已经进过排序的，所以可以在O(n)的时间复杂度内完成。设置一个iterator使其只想nums.begin()+1，并且一直循环到nums.end();并且设置一个计数的量count用来记录当前数字重复出现的次数，初始值设置为1.将上一个数字记录到临时变量t中，如果t=当前指针指向的值时，使count++，表示重复出现，如果不等于则将count恢复到初始值；当count的值出现了3次以上的情况下，将当前it指针对应的值从nums中删除，并且最终返回nums的大小，题目得解

代码如下：

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
    	int count=1;
    	int number=nums.size();
    	if(number==0) return 0;
	vector<int>::iterator it;
	it=nums.begin();
	int t =*it;
	it++;

	for(it; it!=nums.end();it++)
	{
		if(t==*it) count++;
		else if(t!=*it)count=1;
		if(count>=3) 
		{
			it=nums.erase(it);
			it--;
		}
		t=*it;
	}
		int size=nums.size();

	return size;
    }
};