难度:中等
题目描述
给你一个有序数组 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 。 不需要考虑数组中超出新长度后面的元素。
思路
这道题说难也不难,但是我测试了十几次才过,主要就是小的知识点一定要注意到。解题方法类似于“删除哦爱旭数组中的重复项”这道题。依然可以想象成使用两个指针,但是需要注意的是,这道题目要求是出现次数在2次以上的元素需要让它们只出现两次,因此这次可以想象成一个二级指针,每次指向的是当前元素的前面第二个元素,判断是否一样,若不相同则可以覆盖当前元素。
需要仔细注意的是:
这里需要确保数组元素超过两个才能进入for循环,否则若n<=2时进入for循环会出现访问越界的情况(在for循环中会出现这种情况:i=2且i<2,很明显这种情况是错误的)。
代码
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int n=nums.size();
int m=2;
if(n<=2) return n;
else{
for(int i=2;i<n;i++){
if(nums[i]!=nums[m-2]){
nums[m]=nums[i];
m++;
}
}
}
return m;
}
};
依然要注意,int型函数必须有返回值。
复杂度
-
时间复杂度:O(n),其中 n 是数组的长度。我们只需要遍历该数组一次。
-
空间复杂度:O(1)。我们只需要常数的空间存储若干变量。