LintCode_100 删除排序数组中的重复数字 ||

题目

跟进“删除重复数字”：

如果可以允许出现两次重复将如何处理？

样例

给出数组A =[1,1,1,2,2,3]，你的函数应该返回长度5，此时A=[1,1,2,2,3]。

C++代码

 1 int removeDuplicates(vector<int> &nums) {
 2         // write your code here
 3         if(nums.size() == 0) return 0;
 4         int i,j;
 5         int count = 1;
 6         for(i = 1,j = 1; i < nums.size(); ++i)
 7         {
 8             if(nums[i] != nums[j-1])
 9             {
10                 nums[j++] = nums[i];
11                 count = 0;
12             }
13             else if(count < 2 && nums[i] == nums[j-1])
14             {
15                 nums[j++] = nums[i];
16             }
17             count++;
18         }
19 }