【中等题】
题目:
给你一个有序数组 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()){// 区间内元素相等,且区间大小小于等于2if ((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数组长度减1nums.pop_back();}// 更新左边界else if (nums[right] != nums[left]) {left = right;}}return nums.size();}
};
运行结果
把while判断条件改成:
while(right < nums.size())
运行结果明显变好: