力扣LeetCode第80题 删除有序数组中的重复项 II

?一、题目

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

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

示例 1:
给定 nums = [1,1,1,2,2,3],
函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 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 。
你不需要考虑数组中超出新长度后面的元素。

二、思路及图解

1、思路

• 使用双指针解决
• 如果数组 nums的长度等于 2，则数组一定符合，直接返回 数组长度。
• 设置两个指针，slow指针和j指针，slow指针可以当做新数组的指针，而j指针?为原数组依次遍历的指针。
• 因为是满足最多两个重复，则可以推广为，最多满足k个重复。

2、图解

• j 表示快指针，slow表示慢指针；初始时快指针指向下标为2，慢指针指向1。
• 若为k的话，则slow指针应指向k-1，j 指向k
• 如下图：?

?

• ?第一次while循环，先判断j指针所指向元素是否等于slow的前一个元素，也就是新数组的倒数第二个元素，因为指向倒数第二个元素则满足可以保留2个重复元素。则与slow-1比较
• 若为k的话，则与倒数第k个元素比较，则相当于与slow-k+1比较
• 比较后相等，j 指针向右移动一位，

????????

• 第二次循环，j 所指元素与 倒数第二个元素不同 则++slow 并赋值?

????????

????????

• 依次类推，遍历结束之后，新的长度即为slow + 1，返回slow+1即可。

三、题目代码解析

1、代码

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        if (nums.size() <= 2) {
            return nums.size();
        }
        int slow = 1;
        for (int j = 2; j < nums.size(); j++) {
            if (nums[j] != nums[slow - 1]) {
                nums[++slow] = nums[j];
            }
        }
        return slow + 1;
    }
};