Problem:
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.
Analysis:
Solutions:
C++:
int removeDuplicates(vector<int>& nums)
{
if(nums.size() <= 2)
return nums.size();
for(vector<int>::iterator it = nums.begin(); it != nums.end();) {
if(it == nums.begin() || (it != nums.end() - 1 && *it != *(it - 1))) {
if(*it != *(it + 1)) {
++it;
continue;
} else {
++it;
if(*it != *(it + 1)) {
++it;
continue;
} else {
for(++it; it != nums.end() && *it == *(it - 1);) {
it = nums.erase(it);
}
}
}
} else
++it;
}
return nums.size();
}
Java
:
Python: