80. 删除有序数组中的重复项 II【 力扣(LeetCode) 】

一、题目描述

  给你一个有序数组 nums ，请你 原地 删除重复出现的元素，使得出现次数超过两次的元素只出现两次 ，返回删除后数组的新长度。

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

二、测试用例

示例 1：

输入：nums = [1,1,1,2,2,3]
输出：5, nums = [1,1,2,2,3]
解释：函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3。 不需要考虑数组中超出新长度后面的元素。

示例 2：

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

三、解题思路

1. 基本思路：序列已经按照非严格递增的顺序排序，则重复的元素一定连续，则考虑用不需要删除的元素来填充需要删除的元素，来避免大量元素的移动。
2. 具体思路：序列前两个元素一定不用删除，维护两个指针 i 和 k ，其中 i 初始化为 2，用于遍历序列，寻找不需要删除的元素；k 初始化为 1 ，用于存放不需要删除的元素。
   • 因为重复的元素一定连续，则如果指针 i 和指针 k 指向的元素不相同，说明该元素还没有保存，则指针 i 所指的元素是不用删除的，则将其保存给指针 k 。
   • 如果指针 i 和指针 k 指向的元素相同：
     • 如果指针 i 和指针 k-1 指向的元素不相同，说明该元素只保存了一次，所以该元素是不需要删除的，则将其保存给指针 k 。
     • 如果指针 i 和指针 k-1 指向的元素相同，说明该元素已经保存了两次了，则该元素需要删除。
   • 重复上述操作，直至遍历完序列为止。
3. 扩展：需要删除重复元素且元素最多只能保存出现 x 次，则指针 i 与指针 k，k-1，… ，k-x+1 所指的元素不相同都是不需要删除的元素。【不知道算不算这一类问题的通解，嘿嘿！】

四、参考代码

时间复杂度：$O(n)$
空间复杂度：$O(1)$

int removeDuplicates(vector<int>& nums) {
    int n=nums.size();
    if(n<=2) return n;
    int k=1;
    for(int i=2;i<n;i++){
        if(nums[i]!=nums[k]||nums[i]!=nums[k-1])
            nums[++k]=nums[i];
    }
    return k+1;
}

测试结果：