【leetcode】80. 删除排序数组中的重复项 II

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

不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

示例 1:

【leetcode】80. 删除排序数组中的重复项 II

 

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;
    }
};