题目:
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
思路:
由于是有序数组,所以可以一次遍历数组,通过int变量len存储已经遍历到的非重复元素个数,通过nums[len-2]访问到倒数第二个元素,如果与倒数第二个元素相同,那么就一定与倒数第一个元素相同,即已经选出两个与该元素相同的元素了。
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if(nums.empty())
return 0;
int len; //当前数组中已经筛选出的非重复元素个数
len = 1; //数组第一个个元素直接计数
for(int i = 1;i < nums.size();i++){
if(i == 1 || nums[i] != nums[len -2]){ //数组第二个元素,或有等于或少于一个重复元素被选出的元素
nums[len] = nums[i];
++len;
}
}
return len;
}
};