LeetCode 80. 删除排序数组中的重复项 II(C++)

题目:

给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。

不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
LeetCode 80. 删除排序数组中的重复项 II(C++)_第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;
    }
};

你可能感兴趣的:(LeetCode,c++)