给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
示例 1:
给定 nums = [1,1,1,2,2,3], 函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。 你不需要考虑数组中超出新长度后面的元素。
示例 2:
给定 nums = [0,0,1,1,1,1,2,3,3], 函数应返回新长度 length = 7, 并且原数组的前五个元素被修改为 0, 0, 1, 1, 2, 3, 3 。 你不需要考虑数组中超出新长度后面的元素。
错误:
class Solution {
public:
int removeDuplicates(vector<int>& nums)
{
int len = nums.size();
if(len <= 2)
return len;
int pos = 2;
for(int i = 2; i < len; i++)
{
if(nums[i] == nums[i - 1] && nums[i] == nums[i - 2])
{
continue;
}
else
{
nums[pos++] = nums[i];
}
}
return pos;
}
};
如果是111223这种情况的话,如果按照上面算法的情况,把把2赋值给前面重复的位置就变成了112223,这样到最后结果就为1123,判定了2的重复次数大于2.但是2的次数并没有大于2。所以要在上面算法的情况下新加一个判定,再设置个变量记录最后一个重复的被覆盖的元素。比如11223中的1。
class Solution {
public:
int removeDuplicates(vector<int>& nums)
{
int len = nums.size();
if(len <= 2)
return len;
int pos = 2;
int last = 0;
bool isLast = false;
for(int i = 2; i < len; i++)
{
if(nums[i] == nums[i - 1] && nums[i] == nums[i - 2])
{
if(isLast && i - pos == 1 && nums[i] != last)
{
last = nums[pos];
nums[pos++] = nums[i];
continue;
}
}
else
{
isLast = true;
last = nums[pos];
nums[pos++] = nums[i];
}
}
return pos;
}
};