一、题目
给你一个有序数组
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;
}
};
