给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
示例 1:
解题思路一:
题解思路:利用两个指针从索引1开始遍历,判断j位置和前面一个位置的值是否相同,再判断相同次数是否超过2,如果未超过2,则将j上的值赋值给k上,k++后移,time也+1,如果超过2,则k不动,j++后移;如果j位置和前面一个位置的值不相同,则将time赋值为1,并且将j上的值赋值给k上,k++后移
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if(nums.size()<=2){
return nums.size();
}
int time=1;
int k=1;
for(int j=1;j<nums.size();j++){
if(nums[j]==nums[j-1]){
if(time<2){
nums[k++]=nums[j];
time++;
}
}else{
time=1;
nums[k++]=nums[j];
}
}
return k;
}
};
解法二(更快更优):
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if(nums.size()<=2){
return nums.size();
}
//保证k指针指向的始终是正确的符合要求的数字
int k=0,i=0;
for(int i=0;i<nums.size();i++){
if(k<2||nums[i]!=nums[k-2]){
if(i!=k){
nums[k++]=nums[i];
}else{
k++;
}
}
}
return k;
}
};