给定一个排序数组,你需要在 原地 删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
一、单指针遍历求解(耗时180ms)
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if (nums.empty()) {
return 0;
}
//遍历指针
for (auto tail = nums.begin(); tail != nums.end()-1;) {
if (*tail == *(tail + 1)) {
nums.erase(tail + 1);
continue;
}
tail++;
}
return nums.size();
}
};
二、双指针求解(耗时12ms)
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if (nums.empty()) {
return 0;
}
auto pre = nums.begin();
auto tail = nums.begin();
//遍历移动双指针
for (; tail != nums.end(); tail++) {
if (*tail == *pre) {
continue;
}
pre++;
if (tail != pre) {
*pre = *tail;
}
}
//删除
nums.erase(pre+1, tail);
return nums.size();
}
};