【中等题】
题目:
给你一个有序数组 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:
输入: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。不需要考虑数组中超出新长度后面的元素。
思路:
【双指针】维护一个区间,left作为左边界,right作为有边界。如果区间大小大于2,就移动数组。如果right元素与left不同的时候,就让 left变成right继续往右走。
代码:
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
int left = 0, right = 0;
while(left < nums.size() && right < nums.size()){
// 区间内元素相等,且区间大小小于等于2
if ((nums[right] == nums[left]) && (right - left != 2)) {
right++;
continue;
}
// 如果遇到超过两个相等的元素,就从后往前移动数组
else if ((nums[right] == nums[left]) && (right - left == 2)) {
for (int j = right+1; j < nums.size(); j++) {
nums[j - 1] = nums[j];
}
// 删除尾元素,vector数组长度减1
nums.pop_back();
}
// 更新左边界
else if (nums[right] != nums[left]) {
left = right;
}
}
return nums.size();
}
};
运行结果
把while判断条件改成:
while(right < nums.size())
运行结果明显变好: