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

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

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

示例 1:

给定 nums = [1,1,1,2,2,3],函数应返回新长度 length = , 并且原数组的前五个元素被修改为 1, 1, 2, 2,3 。

你不需要考虑数组中超出新长度后面的元素。

示例 2:

给定 nums = [0,0,1,1,1,1,2,3,3],函数应返回新长度 length =7, 并且原数组的前五个元素被修改为 0, 0, 1, 1, 2, 3, 3 。
你不需要考虑数组中超出新长度后面的元素。

思路

双指针遍历+计数器。计数器>2就只移动p2,计数器<2就判断情况

code

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        if(nums.size()<2)
            return nums.size();
        int len = nums.size();
        int p1=0,p2=1;
        int cnt = 1;
        while(p2<len){
            if(nums[p1]==nums[p2] && cnt < 2){
                nums[++p1]=nums[p2];
                ++cnt;
            }
            else if(nums[p1]!=nums[p2]){
                nums[++p1]=nums[p2];
                cnt=1;
            }
            p2++;
        }
        return p1+1;
    }
};