题目链接:80. 删除排序数组中的重复项 II
给定一个增序排列数组 nums ,你需要在 原地 删除重复出现的元素,使得每个元素最多出现两次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 \(O(1)\) 额外空间的条件下完成。
示例 1:
输入:nums = [1,1,1,2,2,3]
输出:5, nums = [1,1,2,2,3]
解释:函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。 你不需要考虑数组中超出新长度后面的元素。
目的是一次 for
循环找到一个元素项,快指针在数组中利用 while
循环找到最后一个相同的元素项。如果 num[i] 和 num[i-1]
相同的话,就进行两次快慢指针赋值操作。如果不相同就进行一次赋值操作。
程序有 bug
:
while
循环要与 num[i+1]
比较,所以要考虑数组访问越界错误。num[i] 和 num[i-1]
比较,所以要对 i == 0
进行特判class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int n = nums.size();
int j = 0;
for (int i = 0; i < n; i++) {
if (i == n - 1) {
nums[j++] = nums[i];
break;
}
while (i != n - 1 && nums[i] == nums[i + 1]) {
i++;
}
if (i == 0) {
nums[j++] = nums[i];
continue;
}
if (nums[i] == nums[i-1]) {
nums[j++] = nums[i];
nums[j++] = nums[i];
} else {
nums[j++] = nums[i];
}
}
return j;
}
};
本人思路没有考虑 快指针 与 慢指针 之间的关系,而只是让 快指针 充当寻找元素的作用,让 慢指针 充当赋值元素的作用。
大佬写的代码:既优雅,还能双百??
考虑最后数组的存储形式:[0,0,1,1,2,3,3]
也就是说,快指针 可以只和 慢指针 相比较。如果 慢指针 已经赋值两个元素了,则之后的相同元素不应该继续赋值。
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int len = 0;
for (int i = 0; i < nums.size(); i++) {
if (len < 2 || nums[i] != nums[len-2]) //注意len < 2特判
nums[len++] = nums[i];
}
return len;
}
};
原文:https://www.cnblogs.com/suyifan/p/14457173.html