leetcode80. Remove Duplicates from Sorted List II

Posted 2021-02-05 JESSET

题目说明

给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素最多出现两次，返回移除后数组的新长度。
https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array-ii/

解法1

时间复杂度：O(n)
空间复杂度：O(1)
思路：使用双指针，第一个指针j表示调整后的数组最后一个位置，第二个指针i遍历原数组。
当nums[i]与nums[j]不等时，直接添加即可，将i位置添加到新数组末尾，并记录下匹配次数（此时为1）。
当nums[i]与nums[j]相等时，就需要判断匹配次数了，判断是否满足匹配次数小于2，若满足则将位置添加到新数组末尾，匹配次数加1；不满足，则直接跳过。
若要修改最大出现的次数，修改max值即可。

int removeDuplicates(vector<int>& nums) 
{
    if (nums.empty())
        return 0;
    int j = 0;
    int count = 1;
    int max = 2;
    for (int i = 1; i < nums.size(); i ++)
    {
        if (nums[i] == nums[j] && count < max)
        {
            nums[++j] = nums[i];
            count ++;
        }
        else if (nums[i] != nums[j])
        {
            nums[++j] = nums[i];
            count = 1;
        }
    }
    return j + 1;
}