题目
给你一个有序数组
nums
,请你 原地 删除重复出现的元素,使得出现次数超过两次的元素只出现两次 ,返回删除后数组的新长度。不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
说明:
为什么返回数值是整数,但输出的答案是数组呢?
请注意,输入数组是以「引用」方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。
你可以想象内部操作如下:
// nums 是以“引用”方式传递的。也就是说,不对实参做任何拷贝
int len = removeDuplicates(nums);
// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。
for (int i = 0; i < len; i++) {
print(nums[i]);
}
示例 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。不需要考虑数组中超出新长度后面的元素。
题解
- 指针的使用: 使用两个指针,
fast
和slow
,来遍历和修改向量。 - 遍历过程: 通过
fast
指针迭代整个向量。当fast
指针找到一个不同于前一个元素的元素时,将其复制到slow
指针的位置,然后增加slow
指针。 - 原地修改: 由于题目要求原地删除重复元素,所以没有使用额外的数据结构。数组中的重复元素被覆盖,而不是通过创建新的数组来存储唯一元素。
- 返回值: 函数返回
slow
指针的值,它表示修改后向量的长度,即不包含重复元素的向量的长度。 - 特殊情况处理: 在处理空向量的情况下,首先检查向量的大小是否为零。如果是零,则直接返回零,因为没有重复元素需要删除。
代码
class Solution { public: int removeDuplicates(vector<int>& nums) { int n = nums.size(); if (n <= 2){ return n; } int slow = 2,fast = 2; while(fast < n){ if (nums[slow - 2] != nums[fast]){ nums[slow] =nums[fast]; slow++; } fast++; } return slow; } };