【leetcode】80. 删除排序数组中的重复项 II
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
示例 1:
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();
}
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;
}
};