【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; } };