题目:给你一个有序数组 nums ,请你 原地 删除重复出现的元素,使得出现次数超过两次的元素只出现两次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
#include
#include
using namespace std;
class Solution
{
public:
int removeDuplicates(vector<int>& nums)
{
for (auto itr = nums.begin(); itr != nums.end() - 2;)
{
if (*itr == *(itr + 1))
{
itr++;
while (*itr == *(itr + 1))
{
itr = nums.erase(itr);
}
}
else
{
itr++;
}
}
for (auto& itr : nums)
{
cout << itr << endl;
}
return nums.size();
}
};
int main()
{
vector<int> nums = {1,2,2,2,2,3,3,4};
Solution a;
a.removeDuplicates(nums);
return 0;
}