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